Pedro Cuenca 0343d8f531 Do not use torch.float64 on the mps device (#942) ...

* Add failing test for #940.

* Do not use torch.float64 in mps.

* style

* Temporarily skip add_noise for IPNDMScheduler.

Until #990 is addressed.

2022-10-26 11:56:43 +02:00

..

fixtures/custom_pipeline

Custome Pipelines (#744)

2022-10-06 16:54:02 +02:00

pipelines

[Dance Diffusion] Better naming (#981)

2022-10-25 19:52:41 +02:00

repo_utils

Introduce the copy mechanism (#924)

2022-10-20 20:26:03 +02:00

__init__.py

fix issues with loading, add test for pipeline

2022-06-07 15:40:36 +02:00

conftest.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00

test_config.py

[Tests] Make sure tests are on GPU (#269)

2022-08-29 15:58:11 +02:00

test_layers_utils.py

Add Apple M1 tests (#796)

2022-10-17 20:27:30 +02:00

test_modeling_common_flax.py

Flax: Trickle down norm_num_groups (#789)

2022-10-11 20:05:10 +02:00

test_modeling_common.py

Add Apple M1 tests (#796)

2022-10-17 20:27:30 +02:00

test_models_unet_1d.py

[Dance Diffusion] Add dance diffusion (#803)

2022-10-25 18:39:25 +02:00

test_models_unet_2d.py

[Dance Diffusion] Add dance diffusion (#803)

2022-10-25 18:39:25 +02:00

test_models_vae_flax.py

Flax: Trickle down norm_num_groups (#789)

2022-10-11 20:05:10 +02:00

test_models_vae.py

[MPS] fix mps failing tests (#934)

2022-10-22 09:33:40 +02:00

test_models_vq.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00

test_outputs.py

Fix BaseOutput initialization from dict (#570)

2022-09-20 18:32:16 +02:00

test_pipelines_common.py

Reorganize pipeline tests (#963)

2022-10-24 16:34:01 +02:00

test_pipelines_flax.py

Reorganize pipeline tests (#963)

2022-10-24 16:34:01 +02:00

test_pipelines_onnx_common.py

Reorganize pipeline tests (#963)

2022-10-24 16:34:01 +02:00

test_pipelines.py

Reorganize pipeline tests (#963)

2022-10-24 16:34:01 +02:00

test_scheduler_flax.py

[Flax] added broadcast_to_shape_from_left helper and Scheduler tests (#864)

2022-10-25 13:43:24 +02:00

test_scheduler.py

Do not use torch.float64 on the mps device (#942)

2022-10-26 11:56:43 +02:00

test_training.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00

test_utils.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00