LLM/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2025-12-06 12:34:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
DN6
2025-09-30 12:36:17 +05:30
parent 9938985426
commit a957ea1f36
19 changed files with 24 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/single_file/test_lumina2_transformer.py
View File

@@ -27,7 +27,7 @@ from .single_file_testing_utils import SingleFileModelTesterMixin
enable_full_determinism()
class TestLumina2Transformer2DModelSingleFileTests(SingleFileModelTesterMixin):
class TestLumina2Transformer2DModelSingleFile(SingleFileModelTesterMixin):
model_class = Lumina2Transformer2DModel
ckpt_path = "https://huggingface.co/Comfy-Org/Lumina_Image_2.0_Repackaged/blob/main/split_files/diffusion_models/lumina_2_model_bf16.safetensors"
alternate_keys_ckpt_paths = [

2
tests/single_file/test_model_autoencoder_dc_single_file.py
View File

@@ -32,7 +32,7 @@ from .single_file_testing_utils import SingleFileModelTesterMixin
enable_full_determinism()
class TestAutoencoderDCSingleFileTests(SingleFileModelTesterMixin):
class TestAutoencoderDCSingleFile(SingleFileModelTesterMixin):
model_class = AutoencoderDC
ckpt_path = "https://huggingface.co/mit-han-lab/dc-ae-f32c32-sana-1.0/blob/main/model.safetensors"
repo_id = "mit-han-lab/dc-ae-f32c32-sana-1.0-diffusers"

2
tests/single_file/test_model_controlnet_single_file.py
View File

@@ -29,7 +29,7 @@ from .single_file_testing_utils import SingleFileModelTesterMixin
enable_full_determinism()
class TestControlNetModelSingleFileTests(SingleFileModelTesterMixin):
class TestControlNetModelSingleFile(SingleFileModelTesterMixin):
model_class = ControlNetModel
ckpt_path = "https://huggingface.co/lllyasviel/ControlNet-v1-1/blob/main/control_v11p_sd15_canny.pth"
repo_id = "lllyasviel/control_v11p_sd15_canny"

2
tests/single_file/test_model_flux_transformer_single_file.py
View File

@@ -30,7 +30,7 @@ from .single_file_testing_utils import SingleFileModelTesterMixin
enable_full_determinism()
class TestFluxTransformer2DModelSingleFileTests(SingleFileModelTesterMixin):
class TestFluxTransformer2DModelSingleFile(SingleFileModelTesterMixin):
model_class = FluxTransformer2DModel
ckpt_path = "https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/flux1-dev.safetensors"
alternate_keys_ckpt_paths = ["https://huggingface.co/Comfy-Org/flux1-dev/blob/main/flux1-dev-fp8.safetensors"]

2
tests/single_file/test_model_vae_single_file.py
View File

@@ -32,7 +32,7 @@ from .single_file_testing_utils import SingleFileModelTesterMixin
enable_full_determinism()
class TestAutoencoderKLSingleFileTests(SingleFileModelTesterMixin):
class TestAutoencoderKLSingleFile(SingleFileModelTesterMixin):
model_class = AutoencoderKL
ckpt_path = (
"https://huggingface.co/stabilityai/sd-vae-ft-mse-original/blob/main/vae-ft-mse-840000-ema-pruned.safetensors"

2
tests/single_file/test_model_wan_autoencoder_single_file.py
View File

@@ -27,7 +27,7 @@ from .single_file_testing_utils import SingleFileModelTesterMixin
enable_full_determinism()
class TestAutoencoderKLWanSingleFileTests(SingleFileModelTesterMixin):
class TestAutoencoderKLWanSingleFile(SingleFileModelTesterMixin):
model_class = AutoencoderKLWan
ckpt_path = (
"https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/blob/main/split_files/vae/wan_2.1_vae.safetensors"

4
tests/single_file/test_model_wan_transformer3d_single_file.py
View File

@@ -30,7 +30,7 @@ from .single_file_testing_utils import SingleFileModelTesterMixin
enable_full_determinism()
class TestWanTransformer3DModelText2VideoSingleFileTest(SingleFileModelTesterMixin):
class TestWanTransformer3DModelText2VideoSingleFile(SingleFileModelTesterMixin):
model_class = WanTransformer3DModel
ckpt_path = "https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/blob/main/split_files/diffusion_models/wan2.1_t2v_1.3B_bf16.safetensors"
repo_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
@@ -38,7 +38,7 @@ class TestWanTransformer3DModelText2VideoSingleFileTest(SingleFileModelTesterMix
@require_big_accelerator
class TestWanTransformer3DModelImage2VideoSingleFileTest(SingleFileModelTesterMixin):
class TestWanTransformer3DModelImage2VideoSingleFile(SingleFileModelTesterMixin):
model_class = WanTransformer3DModel
ckpt_path = "https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/blob/main/split_files/diffusion_models/wan2.1_i2v_480p_14B_fp8_e4m3fn.safetensors"
repo_id = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"

2
tests/single_file/test_sana_transformer.py
View File

@@ -11,7 +11,7 @@ from .single_file_testing_utils import SingleFileModelTesterMixin
enable_full_determinism()
class TestSanaTransformer2DModelSingleFileTests(SingleFileModelTesterMixin):
class TestSanaTransformer2DModelSingleFile(SingleFileModelTesterMixin):
model_class = SanaTransformer2DModel
ckpt_path = (
"https://huggingface.co/Efficient-Large-Model/Sana_1600M_1024px/blob/main/checkpoints/Sana_1600M_1024px.pth"

2
tests/single_file/test_stable_diffusion_controlnet_img2img_single_file.py
View File

@@ -28,7 +28,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionControlNetPipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusionControlNetPipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionControlNetPipeline
ckpt_path = (
"https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors"

2
tests/single_file/test_stable_diffusion_controlnet_inpaint_single_file.py
View File

@@ -29,7 +29,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionControlNetInpaintPipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusionControlNetInpaintPipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionControlNetInpaintPipeline
ckpt_path = "https://huggingface.co/botp/stable-diffusion-v1-5-inpainting/blob/main/sd-v1-5-inpainting.ckpt"
original_config = "https://raw.githubusercontent.com/runwayml/stable-diffusion/main/configs/stable-diffusion/v1-inpainting-inference.yaml"

2
tests/single_file/test_stable_diffusion_controlnet_single_file.py
View File

@@ -28,7 +28,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionControlNetPipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusionControlNetPipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionControlNetPipeline
ckpt_path = (
"https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors"

4
tests/single_file/test_stable_diffusion_img2img_single_file.py
View File

@@ -22,7 +22,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionImg2ImgPipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusionImg2ImgPipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionImg2ImgPipeline
ckpt_path = (
"https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors"
@@ -63,7 +63,7 @@ class TestStableDiffusionImg2ImgPipelineSingleFileSlowTests(SDSingleFileTesterMi
@slow
@require_torch_accelerator
class TestStableDiffusion21Img2ImgPipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusion21Img2ImgPipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionImg2ImgPipeline
ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-2-1/blob/main/v2-1_768-ema-pruned.safetensors"
original_config = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml"

4
tests/single_file/test_stable_diffusion_inpaint_single_file.py
View File

@@ -23,7 +23,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionInpaintPipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusionInpaintPipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionInpaintPipeline
ckpt_path = "https://huggingface.co/botp/stable-diffusion-v1-5-inpainting/blob/main/sd-v1-5-inpainting.ckpt"
original_config = "https://raw.githubusercontent.com/runwayml/stable-diffusion/main/configs/stable-diffusion/v1-inpainting-inference.yaml"
@@ -79,7 +79,7 @@ class TestStableDiffusionInpaintPipelineSingleFileSlowTests(SDSingleFileTesterMi
@slow
@require_torch_accelerator
class TestStableDiffusion21InpaintPipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusion21InpaintPipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionInpaintPipeline
ckpt_path = (
"https://huggingface.co/stabilityai/stable-diffusion-2-inpainting/blob/main/512-inpainting-ema.safetensors"

6
tests/single_file/test_stable_diffusion_single_file.py
View File

@@ -27,7 +27,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionPipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusionPipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionPipeline
ckpt_path = (
"https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors"
@@ -87,7 +87,7 @@ class TestStableDiffusionPipelineSingleFileSlowTests(SDSingleFileTesterMixin):
@slow
class TestStableDiffusion21PipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusion21PipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionPipeline
ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-2-1/blob/main/v2-1_768-ema-pruned.safetensors"
original_config = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml"
@@ -120,7 +120,7 @@ class TestStableDiffusion21PipelineSingleFileSlowTests(SDSingleFileTesterMixin):
@nightly
@slow
@require_torch_accelerator
class TestStableDiffusionInstructPix2PixPipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusionInstructPix2PixPipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionInstructPix2PixPipeline
ckpt_path = "https://huggingface.co/timbrooks/instruct-pix2pix/blob/main/instruct-pix2pix-00-22000.safetensors"
original_config = (

2
tests/single_file/test_stable_diffusion_upscale_single_file.py
View File

@@ -24,7 +24,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionUpscalePipelineSingleFileSlowTests(SDSingleFileTesterMixin):
class TestStableDiffusionUpscalePipelineSingleFileSlow(SDSingleFileTesterMixin):
pipeline_class = StableDiffusionUpscalePipeline
ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-x4-upscaler/blob/main/x4-upscaler-ema.safetensors"
original_config = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/x4-upscaling.yaml"

2
tests/single_file/test_stable_diffusion_xl_adapter_single_file.py
View File

@@ -31,7 +31,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionXLAdapterPipelineSingleFileSlowTests(SDXLSingleFileTesterMixin):
class TestStableDiffusionXLAdapterPipelineSingleFileSlow(SDXLSingleFileTesterMixin):
pipeline_class = StableDiffusionXLAdapterPipeline
ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base_1.0.safetensors"
repo_id = "stabilityai/stable-diffusion-xl-base-1.0"

2
tests/single_file/test_stable_diffusion_xl_controlnet_single_file.py
View File

@@ -27,7 +27,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionXLControlNetPipelineSingleFileSlowTests(SDXLSingleFileTesterMixin):
class TestStableDiffusionXLControlNetPipelineSingleFileSlow(SDXLSingleFileTesterMixin):
pipeline_class = StableDiffusionXLControlNetPipeline
ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base_1.0.safetensors"
repo_id = "stabilityai/stable-diffusion-xl-base-1.0"

2
tests/single_file/test_stable_diffusion_xl_img2img_single_file.py
View File

@@ -24,7 +24,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionXLImg2ImgPipelineSingleFileSlowTests(SDXLSingleFileTesterMixin):
class TestStableDiffusionXLImg2ImgPipelineSingleFileSlow(SDXLSingleFileTesterMixin):
pipeline_class = StableDiffusionXLImg2ImgPipeline
ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base_1.0.safetensors"
repo_id = "stabilityai/stable-diffusion-xl-base-1.0"

2
tests/single_file/test_stable_diffusion_xl_single_file.py
View File

@@ -21,7 +21,7 @@ enable_full_determinism()
@slow
@require_torch_accelerator
class TestStableDiffusionXLPipelineSingleFileSlowTests(SDXLSingleFileTesterMixin):
class TestStableDiffusionXLPipelineSingleFileSlow(SDXLSingleFileTesterMixin):
pipeline_class = StableDiffusionXLPipeline
ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base_1.0.safetensors"
repo_id = "stabilityai/stable-diffusion-xl-base-1.0"