Arsalan f427345ab1 Device agnostic testing (#5612) ...

* utils and test modifications to enable device agnostic testing

* device for manual seed in unet1d

* fix generator condition in vae test

* consistency changes to testing

* make style

* add device agnostic testing changes to source and one model test

* make dtype check fns private, log cuda fp16 case

* remove dtype checks from import utils, move to testing_utils

* adding tests for most model classes and one pipeline

* fix vae import

2023-12-05 19:04:13 +05:30

..

__init__.py

[Tests] Improve unet / vae tests (#1018)

2022-10-28 13:43:26 +02:00

test_activations.py

move activation dispatches into helper function (#3656)

2023-06-05 12:30:48 -07:00

test_attention_processor.py

relax tol attention conversion test (#3842)

2023-06-21 12:35:38 -07:00

test_layers_utils.py

Device agnostic testing (#5612)

2023-12-05 19:04:13 +05:30

test_modeling_common_flax.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_modeling_common.py

Device agnostic testing (#5612)

2023-12-05 19:04:13 +05:30

test_models_prior.py

Device agnostic testing (#5612)

2023-12-05 19:04:13 +05:30

test_models_unet_1d.py

Device agnostic testing (#5612)

2023-12-05 19:04:13 +05:30

test_models_unet_2d_condition.py

Device agnostic testing (#5612)

2023-12-05 19:04:13 +05:30

test_models_unet_2d_flax.py

Replace flake8 with ruff and update black (#2279)

2023-02-07 23:46:23 +01:00

test_models_unet_2d.py

Device agnostic testing (#5612)

2023-12-05 19:04:13 +05:30

test_models_unet_3d_condition.py

[LoRA] Centralize LoRA tests (#5086)

2023-09-18 17:54:33 +02:00

test_models_unet_motion.py

Animatediff Proposal (#5413)

2023-11-02 15:04:03 +01:00

test_models_unet_spatiotemporal.py

Add SVD (#5895)

2023-11-29 19:13:36 +01:00

test_models_vae_flax.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_models_vae.py

Device agnostic testing (#5612)

2023-12-05 19:04:13 +05:30

test_models_vq.py

Device agnostic testing (#5612)

2023-12-05 19:04:13 +05:30

test_unet_2d_blocks.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00

test_unet_blocks_common.py

Device agnostic testing (#5612)

2023-12-05 19:04:13 +05:30