huggingface/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-05-28 00:39:35 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

update
Some checks failed
Secret Leaks / trufflehog (push) Has been cancelled
Details

Browse Source
This commit is contained in:
DN6
2026-05-27 16:59:58 +05:30
parent 3814136d2f
commit 7b89516c76
11 changed files with 42 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/lora/test_lora_layers_sd.py
View File

@@ -24,8 +24,6 @@ from safetensors.torch import load_file
from transformers import CLIPTextModel, CLIPTokenizer
from diffusers import (
AutoPipelineForImage2Image,
AutoPipelineForText2Image,
DDIMScheduler,
DiffusionPipeline,
LCMScheduler,
@@ -641,6 +639,8 @@ class LoraIntegrationTests(unittest.TestCase):
release_memory(pipe)
def test_not_empty_state_dict(self):
from diffusers import AutoPipelineForText2Image
# Makes sure https://github.com/huggingface/diffusers/issues/7054 does not happen again
pipe = AutoPipelineForText2Image.from_pretrained(
"stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float16
@@ -655,6 +655,8 @@ class LoraIntegrationTests(unittest.TestCase):
release_memory(pipe)
def test_load_unload_load_state_dict(self):
from diffusers import AutoPipelineForText2Image
# Makes sure https://github.com/huggingface/diffusers/issues/7054 does not happen again
pipe = AutoPipelineForText2Image.from_pretrained(
"stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float16
@@ -705,6 +707,8 @@ class LoraIntegrationTests(unittest.TestCase):
release_memory(pipe)
def test_sdv1_5_lcm_lora_img2img(self):
from diffusers import AutoPipelineForImage2Image
pipe = AutoPipelineForImage2Image.from_pretrained(
"stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float16
)

3
tests/others/test_flashpack.py
View File

@@ -17,7 +17,6 @@ import pathlib
import tempfile
import unittest
from diffusers import AutoPipelineForText2Image
from diffusers.models.auto_model import AutoModel
from ..testing_utils import is_torch_available, require_flashpack, require_torch_gpu
@@ -40,6 +39,8 @@ class FlashPackTests(unittest.TestCase):
@require_flashpack
def test_save_load_pipeline(self):
from diffusers import AutoPipelineForText2Image
pipeline = AutoPipelineForText2Image.from_pretrained(self.model_id)
with tempfile.TemporaryDirectory() as temp_dir:
pipeline.save_pretrained(temp_dir, use_flashpack=True)

6
tests/pipelines/kandinsky3/test_kandinsky3.py
View File

@@ -22,8 +22,6 @@ from PIL import Image
from transformers import AutoConfig, AutoTokenizer, T5EncoderModel
from diffusers import (
AutoPipelineForImage2Image,
AutoPipelineForText2Image,
Kandinsky3Pipeline,
Kandinsky3UNet,
VQModel,
@@ -185,6 +183,8 @@ class Kandinsky3PipelineIntegrationTests(unittest.TestCase):
backend_empty_cache(torch_device)
def test_kandinskyV3(self):
from diffusers import AutoPipelineForText2Image
pipe = AutoPipelineForText2Image.from_pretrained(
"kandinsky-community/kandinsky-3", variant="fp16", torch_dtype=torch.float16
)
@@ -211,6 +211,8 @@ class Kandinsky3PipelineIntegrationTests(unittest.TestCase):
self.assertTrue(np.allclose(image_np, expected_image_np, atol=5e-2))
def test_kandinskyV3_img2img(self):
from diffusers import AutoPipelineForImage2Image
pipe = AutoPipelineForImage2Image.from_pretrained(
"kandinsky-community/kandinsky-3", variant="fp16", torch_dtype=torch.float16
)

3
tests/pipelines/kandinsky3/test_kandinsky3_img2img.py
View File

@@ -23,7 +23,6 @@ from PIL import Image
from transformers import AutoConfig, AutoTokenizer, T5EncoderModel
from diffusers import (
AutoPipelineForImage2Image,
Kandinsky3Img2ImgPipeline,
Kandinsky3UNet,
VQModel,
@@ -210,6 +209,8 @@ class Kandinsky3Img2ImgPipelineIntegrationTests(unittest.TestCase):
backend_empty_cache(torch_device)
def test_kandinskyV3_img2img(self):
from diffusers import AutoPipelineForImage2Image
pipe = AutoPipelineForImage2Image.from_pretrained(
"kandinsky-community/kandinsky-3", variant="fp16", torch_dtype=torch.float16
)

5
tests/pipelines/pag/test_pag_sd.py
View File

@@ -23,7 +23,6 @@ from transformers import CLIPTextConfig, CLIPTextModel, CLIPTokenizer
from diffusers import (
AutoencoderKL,
AutoPipelineForText2Image,
DDIMScheduler,
StableDiffusionPAGPipeline,
StableDiffusionPipeline,
@@ -315,6 +314,8 @@ class StableDiffusionPAGPipelineIntegrationTests(unittest.TestCase):
return inputs
def test_pag_cfg(self):
from diffusers import AutoPipelineForText2Image
pipeline = AutoPipelineForText2Image.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)
@@ -333,6 +334,8 @@ class StableDiffusionPAGPipelineIntegrationTests(unittest.TestCase):
)
def test_pag_uncond(self):
from diffusers import AutoPipelineForText2Image
pipeline = AutoPipelineForText2Image.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)

5
tests/pipelines/pag/test_pag_sd3_img2img.py
View File

@@ -16,7 +16,6 @@ from transformers import (
from diffusers import (
AutoencoderKL,
AutoPipelineForImage2Image,
FlowMatchEulerDiscreteScheduler,
SD3Transformer2DModel,
StableDiffusion3Img2ImgPipeline,
@@ -240,6 +239,8 @@ class StableDiffusion3PAGImg2ImgPipelineIntegrationTests(unittest.TestCase):
return inputs
def test_pag_cfg(self):
from diffusers import AutoPipelineForImage2Image
pipeline = AutoPipelineForImage2Image.from_pretrained(
self.repo_id, enable_pag=True, torch_dtype=torch.float16, pag_applied_layers=["blocks.17"]
)
@@ -268,6 +269,8 @@ class StableDiffusion3PAGImg2ImgPipelineIntegrationTests(unittest.TestCase):
)
def test_pag_uncond(self):
from diffusers import AutoPipelineForImage2Image
pipeline = AutoPipelineForImage2Image.from_pretrained(
self.repo_id, enable_pag=True, torch_dtype=torch.float16, pag_applied_layers=["blocks.(4|17)"]
)

5
tests/pipelines/pag/test_pag_sd_img2img.py
View File

@@ -25,7 +25,6 @@ from transformers import CLIPTextConfig, CLIPTextModel, CLIPTokenizer
from diffusers import (
AutoencoderKL,
AutoencoderTiny,
AutoPipelineForImage2Image,
EulerDiscreteScheduler,
StableDiffusionImg2ImgPipeline,
StableDiffusionPAGImg2ImgPipeline,
@@ -254,6 +253,8 @@ class StableDiffusionPAGImg2ImgPipelineIntegrationTests(unittest.TestCase):
return inputs
def test_pag_cfg(self):
from diffusers import AutoPipelineForImage2Image
pipeline = AutoPipelineForImage2Image.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)
@@ -272,6 +273,8 @@ class StableDiffusionPAGImg2ImgPipelineIntegrationTests(unittest.TestCase):
)
def test_pag_uncond(self):
from diffusers import AutoPipelineForImage2Image
pipeline = AutoPipelineForImage2Image.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)

5
tests/pipelines/pag/test_pag_sd_inpaint.py
View File

@@ -24,7 +24,6 @@ from transformers import CLIPTextConfig, CLIPTextModel, CLIPTokenizer
from diffusers import (
AutoencoderKL,
AutoPipelineForInpainting,
PNDMScheduler,
StableDiffusionPAGInpaintPipeline,
UNet2DConditionModel,
@@ -289,6 +288,8 @@ class StableDiffusionPAGPipelineIntegrationTests(unittest.TestCase):
return inputs
def test_pag_cfg(self):
from diffusers import AutoPipelineForInpainting
pipeline = AutoPipelineForInpainting.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)
@@ -307,6 +308,8 @@ class StableDiffusionPAGPipelineIntegrationTests(unittest.TestCase):
)
def test_pag_uncond(self):
from diffusers import AutoPipelineForInpainting
pipeline = AutoPipelineForInpainting.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)

5
tests/pipelines/pag/test_pag_sdxl.py
View File

@@ -23,7 +23,6 @@ from transformers import CLIPTextConfig, CLIPTextModel, CLIPTextModelWithProject
from diffusers import (
AutoencoderKL,
AutoPipelineForText2Image,
EulerDiscreteScheduler,
StableDiffusionXLPAGPipeline,
StableDiffusionXLPipeline,
@@ -319,6 +318,8 @@ class StableDiffusionXLPAGPipelineIntegrationTests(unittest.TestCase):
return inputs
def test_pag_cfg(self):
from diffusers import AutoPipelineForText2Image
pipeline = AutoPipelineForText2Image.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)
@@ -336,6 +337,8 @@ class StableDiffusionXLPAGPipelineIntegrationTests(unittest.TestCase):
)
def test_pag_uncond(self):
from diffusers import AutoPipelineForText2Image
pipeline = AutoPipelineForText2Image.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)

5
tests/pipelines/pag/test_pag_sdxl_img2img.py
View File

@@ -32,7 +32,6 @@ from transformers import (
from diffusers import (
AutoencoderKL,
AutoPipelineForImage2Image,
EulerDiscreteScheduler,
StableDiffusionXLImg2ImgPipeline,
StableDiffusionXLPAGImg2ImgPipeline,
@@ -304,6 +303,8 @@ class StableDiffusionXLPAGImg2ImgPipelineIntegrationTests(unittest.TestCase):
return inputs
def test_pag_cfg(self):
from diffusers import AutoPipelineForImage2Image
pipeline = AutoPipelineForImage2Image.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)
@@ -321,6 +322,8 @@ class StableDiffusionXLPAGImg2ImgPipelineIntegrationTests(unittest.TestCase):
)
def test_pag_uncond(self):
from diffusers import AutoPipelineForImage2Image
pipeline = AutoPipelineForImage2Image.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)

5
tests/pipelines/pag/test_pag_sdxl_inpaint.py
View File

@@ -33,7 +33,6 @@ from transformers import (
from diffusers import (
AutoencoderKL,
AutoPipelineForInpainting,
EulerDiscreteScheduler,
StableDiffusionXLInpaintPipeline,
StableDiffusionXLPAGInpaintPipeline,
@@ -310,6 +309,8 @@ class StableDiffusionXLPAGInpaintPipelineIntegrationTests(unittest.TestCase):
return inputs
def test_pag_cfg(self):
from diffusers import AutoPipelineForInpainting
pipeline = AutoPipelineForInpainting.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)
@@ -327,6 +328,8 @@ class StableDiffusionXLPAGInpaintPipelineIntegrationTests(unittest.TestCase):
)
def test_pag_uncond(self):
from diffusers import AutoPipelineForInpainting
pipeline = AutoPipelineForInpainting.from_pretrained(self.repo_id, enable_pag=True, torch_dtype=torch.float16)
pipeline.enable_model_cpu_offload(device=torch_device)
pipeline.set_progress_bar_config(disable=None)