Pedro Cuenca af279434d0 Flax tests: don't hardcode number of devices (#1175) ...

Flax tests: don't hardcode number of devices.

This makes it possible to test on CPU/GPU. However, expected slices are
only checked when there are 8 devices.

2022-11-09 20:04:43 +01:00

..

fixtures/custom_pipeline

[Scheduler] Move predict epsilon to init (#1155)

2022-11-08 18:08:08 +01:00

models

MPS schedulers: don't use float64 (#1169)

2022-11-08 13:11:33 +01:00

pipelines

Add LDM Super Resolution pipeline (#1116)

2022-11-09 13:42:16 +01:00

repo_utils

Move accelerate to a soft-dependency (#1134)

2022-11-04 14:58:52 +01: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

[Scheduler] Move predict epsilon to init (#1155)

2022-11-08 18:08:08 +01:00

test_layers_utils.py

VQ-diffusion (#658)

2022-11-03 16:10:28 +01: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_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

Flax tests: don't hardcode number of devices (#1175)

2022-11-09 20:04:43 +01:00

test_pipelines_onnx_common.py

Reorganize pipeline tests (#963)

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

test_pipelines.py

[Loading] Make sure loading edge cases work (#1192)

2022-11-09 12:28:56 +01:00

test_scheduler_flax.py

[CI] Framework and hardware-specific CI tests (#997)

2022-11-02 14:07:07 +01:00

test_scheduler.py

Fix slow tests (#1210)

2022-11-09 11:22:12 +01: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