dg845 4c8a05f115 Fix Consistency Models UNet2DMidBlock2D Attention GroupNorm Bug (#4863) ...

* Add attn_groups argument to UNet2DMidBlock2D to control theinternal Attention block's GroupNorm.

* Add docstring for attn_norm_num_groups in UNet2DModel.

* Since the test UNet config uses resnet_time_scale_shift == 'scale_shift', also set attn_norm_num_groups to 32.

* Add test for attn_norm_num_groups to UNet2DModelTests.

* Fix expected slices for slow tests.

* Also fix tolerances for slow tests.

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

2023-09-15 11:27:51 +01:00

..

__init__.py

Fix conversion script

2022-07-15 17:00:41 +00:00

change_naming_configs_and_checkpoints.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

conversion_ldm_uncond.py

Replace flake8 with ruff and update black (#2279)

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

convert_asymmetric_vqgan_to_diffusers.py

Asymmetric vqgan (#3956)

2023-07-20 17:51:06 +02:00

convert_consistency_to_diffusers.py

Fix Consistency Models UNet2DMidBlock2D Attention GroupNorm Bug (#4863)

2023-09-15 11:27:51 +01:00

convert_dance_diffusion_to_diffusers.py

Replace flake8 with ruff and update black (#2279)

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

convert_ddpm_original_checkpoint_to_diffusers.py

Ruff: apply same rules as in transformers (#2827)

2023-03-27 16:18:57 +02:00

convert_diffusers_to_original_sdxl.py

add convert diffuser pipeline of XL to original stable diffusion (#4596)

2023-08-22 19:11:06 +05:30

convert_diffusers_to_original_stable_diffusion.py

Replace flake8 with ruff and update black (#2279)

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

convert_dit_to_diffusers.py

Replace flake8 with ruff and update black (#2279)

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

convert_gligen_to_diffusers.py

Add GLIGEN Text Image implementation (#4777)

2023-09-01 15:48:01 +05:30

convert_if.py

add model (#3230)

2023-04-25 14:20:43 -07:00

convert_k_upscaler_to_diffusers.py

Replace flake8 with ruff and update black (#2279)

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

convert_kakao_brain_unclip_to_diffusers.py

Replace flake8 with ruff and update black (#2279)

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

convert_kandinsky_to_diffusers.py

update conversion script for Kandinsky unet (#3766)

2023-06-14 06:57:53 -10:00

convert_ldm_original_checkpoint_to_diffusers.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

convert_lora_safetensor_to_diffusers.py

make style

2023-03-06 10:51:03 +00:00

convert_models_diffuser_to_diffusers.py

Ruff: apply same rules as in transformers (#2827)

2023-03-27 16:18:57 +02:00

convert_ms_text_to_video_to_diffusers.py

[MS Text To Video] Add first text to video (#2738)

2023-03-22 18:39:33 +01:00

convert_music_spectrogram_to_diffusers.py

Music Spectrogram diffusion pipeline (#1044)

2023-03-23 14:06:17 +01:00

convert_ncsnpp_original_checkpoint_to_diffusers.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

convert_original_audioldm2_to_diffusers.py

Add AudioLDM 2 (#4549)

2023-08-21 12:34:21 +01:00

convert_original_audioldm_to_diffusers.py

Remove code snippets containing is_safetensors_available() (#4521)

2023-08-11 11:05:22 +05:30

convert_original_controlnet_to_diffusers.py

fix some typo error (#4546)

2023-08-10 06:49:25 +05:30

convert_original_musicldm_to_diffusers.py

Convert MusicLDM (#4579)

2023-08-25 13:31:00 +01:00

convert_original_stable_diffusion_to_diffusers.py

allow loading of sd models from safetensors without online lookups using local config files (#5019)

2023-09-14 12:30:15 +02:00

convert_original_t2i_adapter.py

t2i pipeline (#3932)

2023-07-17 12:55:44 -07:00

convert_shap_e_to_diffusers.py

Shap-E: add support for mesh output (#4062)

2023-07-20 18:05:13 +02:00

convert_stable_diffusion_checkpoint_to_onnx.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

convert_stable_diffusion_controlnet_to_onnx.py

Convert Stable Diffusion ControlNet to TensorRT (#4465)

2023-08-11 08:12:26 +05:30

convert_stable_diffusion_controlnet_to_tensorrt.py

Convert Stable Diffusion ControlNet to TensorRT (#4465)

2023-08-11 08:12:26 +05:30

convert_tiny_autoencoder_to_diffusers.py

Remove code snippets containing is_safetensors_available() (#4521)

2023-08-11 11:05:22 +05:30

convert_unclip_txt2img_to_image_variation.py

Replace flake8 with ruff and update black (#2279)

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

convert_unidiffuser_to_diffusers.py

[WIP] Add UniDiffuser model and pipeline (#2963)

2023-05-26 17:27:30 +05:30

convert_vae_diff_to_onnx.py

make style

2023-03-06 10:40:18 +00:00

convert_vae_pt_to_diffusers.py

[Conversion] Small fixes (#3848)

2023-06-22 13:52:59 +02:00

convert_versatile_diffusion_to_diffusers.py

Ruff: apply same rules as in transformers (#2827)

2023-03-27 16:18:57 +02:00

convert_vq_diffusion_to_diffusers.py

Replace flake8 with ruff and update black (#2279)

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

convert_wuerstchen.py

Wuerstchen fixes (#4942)

2023-09-11 15:47:53 +02:00

convert_zero123_to_diffusers.py

[Pipelines] Add community pipeline for Zero123 (#4295)

2023-08-02 19:36:49 +02:00

generate_logits.py

Replace flake8 with ruff and update black (#2279)

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