Sayak Paul c1926cef6b [tests] refactor vae tests (#9808) ...

* add: autoencoderkl tests

* autoencodertiny.

* fix

* asymmetric autoencoder.

* more

* integration tests for stable audio decoder.

* consistency decoder vae tests

* remove grad check from consistency decoder.

* cog

* bye test_models_vae.py

* fix

* fix

* remove allegro

* fixes

* fixes

* fixes

---------

Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>

2024-12-04 15:58:36 +05:30

..

allegro

[core] Allegro T2V (#9736)

2024-10-29 13:14:36 +05:30

amused

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

animatediff

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

audioldm

[Tests] reduce the model size in the audioldm fast test (#7833)

2024-05-02 06:03:52 +05:30

audioldm2

Revert "[LoRA] fix: lora loading when using with a device_mapped mode… (#9823)

2024-10-31 08:19:32 -10:00

aura_flow

[CI] Multiple Slow Test fixes. (#9198)

2024-08-19 13:31:09 +05:30

blipdiffusion

[Tests] reduce the model size in the blipdiffusion fast test (#7849)

2024-05-03 07:46:48 +05:30

cogvideo

[tests] fix name and unskip CogI2V integration test (#9683)

2024-10-16 16:28:19 +05:30

cogview3

CogView3Plus DiT (#9570)

2024-10-14 19:30:36 +05:30

consistency_models

Memory clean up on all Slow Tests (#7514)

2024-03-29 14:23:28 +05:30

controlnet

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

controlnet_flux

Fix prepare latent image ids and vae sample generators for flux (#9981)

2024-11-21 13:02:31 +05:30

controlnet_hunyuandit

[Tencent Hunyuan Team] Add Hunyuan-DiT ControlNet Inference (#8694)

2024-06-26 00:43:03 -10:00

controlnet_sd3

Improve control net block index for sd3 (#9758)

2024-11-20 12:45:18 -10:00

controlnet_xs

[tests] refactor vae tests (#9808)

2024-12-04 15:58:36 +05:30

dance_diffusion

Memory clean up on all Slow Tests (#7514)

2024-03-29 14:23:28 +05:30

ddim

[Tests] reduce the model size in the ddim fast test (#7803)

2024-04-30 07:54:38 +05:30

ddpm

[Tests] reduce the model size in the ddpm fast test (#7797)

2024-04-30 08:11:03 +05:30

deepfloyd_if

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

dit

[Core] Introduce class variants for Transformer2DModel (#7647)

2024-05-31 13:40:27 +05:30

flux

Flux Fill, Canny, Depth, Redux (#9985)

2024-11-23 01:41:25 -10:00

hunyuan_dit

[Core] fix QKV fusion for attention (#8829)

2024-07-24 06:52:19 +05:30

i2vgen_xl

[Tests] fix slices of 26 tests (first half) (#8959)

2024-07-25 14:56:49 +05:30

ip_adapters

a few fix for SingleFile tests (#9522)

2024-09-24 13:36:53 -10:00

kandinsky

Revert "[LoRA] fix: lora loading when using with a device_mapped mode… (#9823)

2024-10-31 08:19:32 -10:00

kandinsky2_2

Revert "[LoRA] fix: lora loading when using with a device_mapped mode… (#9823)

2024-10-31 08:19:32 -10:00

kandinsky3

[Tests] Speed up fast pipelines part II (#7521)

2024-04-02 13:24:56 +05:30

kolors

[Kolors] Add PAG (#8934)

2024-08-07 09:29:52 +05:30

latent_consistency_models

[tests] remove/speedup some low signal tests (#9285)

2024-09-03 13:59:18 +05:30

latent_diffusion

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

latte

[tests] fix broken xformers tests (#9206)

2024-08-22 15:17:47 +05:30

ledits_pp

a few fix for SingleFile tests (#9522)

2024-09-24 13:36:53 -10:00

lumina

[Tests] Reduce the model size in the lumina test (#8985)

2024-09-23 20:35:50 +05:30

marigold

[Marigold tests] add is_flaky decorator to some Marigold tests (#8696)

2024-06-25 06:27:28 -10:00

mochi

[core] Mochi T2V (#9769)

2024-11-05 20:33:41 +05:30

musicldm

Revert "[LoRA] fix: lora loading when using with a device_mapped mode… (#9823)

2024-10-31 08:19:32 -10:00

pag

Add StableDiffusion3PAGImg2Img Pipeline + Fix SD3 Unconditional PAG (#9932)

2024-12-02 21:39:47 -10:00

paint_by_example

Memory clean up on all Slow Tests (#7514)

2024-03-29 14:23:28 +05:30

pia

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

pixart_alpha

[Core] Introduce class variants for Transformer2DModel (#7647)

2024-05-31 13:40:27 +05:30

pixart_sigma

[Core] add QKV fusion to AuraFlow and PixArt Sigma (#8952)

2024-08-05 14:09:37 -10:00

pndm

[Tests] Update a deprecated parameter in test files and fix several typos (#7277)

2024-03-14 12:17:35 -07:00

semantic_stable_diffusion

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

shap_e

[Tests] fix slices of 26 tests (first half) (#8959)

2024-07-25 14:56:49 +05:30

stable_audio

[CI] More fixes for Fast GPU Tests on main (#9300)

2024-09-02 17:51:48 +05:30

stable_cascade

Revert "[LoRA] fix: lora loading when using with a device_mapped mode… (#9823)

2024-10-31 08:19:32 -10:00

stable_diffusion

a few fix for SingleFile tests (#9522)

2024-09-24 13:36:53 -10:00

stable_diffusion_2

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

stable_diffusion_3

Test skip_guidance_layers in SD3 pipeline (#10102)

2024-12-03 14:28:31 -10:00

stable_diffusion_adapter

Revert "[LoRA] fix: lora loading when using with a device_mapped mode… (#9823)

2024-10-31 08:19:32 -10:00

stable_diffusion_gligen

add a from_pipe method to DiffusionPipeline (#7241)

2024-04-01 13:02:00 -10:00

stable_diffusion_gligen_text_image

[Tests] Speed up fast pipelines part II (#7521)

2024-04-02 13:24:56 +05:30

stable_diffusion_image_variation

[Tests] fix slices of 26 tests (first half) (#8959)

2024-07-25 14:56:49 +05:30

stable_diffusion_k_diffusion

Memory clean up on all Slow Tests (#7514)

2024-03-29 14:23:28 +05:30

stable_diffusion_ldm3d

Memory clean up on all Slow Tests (#7514)

2024-03-29 14:23:28 +05:30

stable_diffusion_panorama

Optimize test files by fixing CPU-offloading usage (#8409)

2024-06-06 09:51:26 -10:00

stable_diffusion_safe

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

stable_diffusion_sag

add a from_pipe method to DiffusionPipeline (#7241)

2024-04-01 13:02:00 -10:00

stable_diffusion_xl

Revert "[LoRA] fix: lora loading when using with a device_mapped mode… (#9823)

2024-10-31 08:19:32 -10:00

stable_unclip

Revert "[LoRA] fix: lora loading when using with a device_mapped mode… (#9823)

2024-10-31 08:19:32 -10:00

stable_video_diffusion

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

text_to_video_synthesis

make pipelines tests device-agnostic (part1) (#9399)

2024-11-22 15:32:54 +05:30

unclip

Optimize test files by fixing CPU-offloading usage (#8409)

2024-06-06 09:51:26 -10:00

unidiffuser

Revert "[LoRA] fix: lora loading when using with a device_mapped mode… (#9823)

2024-10-31 08:19:32 -10:00

wuerstchen

Revert "[LoRA] fix: lora loading when using with a device_mapped mode… (#9823)

2024-10-31 08:19:32 -10:00

__init__.py

Reorganize pipeline tests (#963)

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

pipeline_params.py

AnimateDiff Video to Video (#6328)

2024-01-24 18:22:26 +05:30

test_pipeline_utils.py

Improve downloads of sharded variants (#9869)

2024-11-08 17:49:00 +05:30

test_pipelines_auto.py

a few fix for SingleFile tests (#9522)

2024-09-24 13:36:53 -10:00

test_pipelines_combined.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_pipelines_common.py

[tests] refactor vae tests (#9808)

2024-12-04 15:58:36 +05:30

test_pipelines_flax.py

Discourage using deprecated revision parameter (#8573)

2024-06-24 10:06:49 -07:00

test_pipelines_onnx_common.py

[Tests] parallelize (#3078)

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

test_pipelines.py

Fix bug in Textual Inversion Unloading (#9304)

2024-10-19 02:37:32 -10:00