diff --git a/tests/lora/test_lora_layers_sd.py b/tests/lora/test_lora_layers_sd.py index 933bf2336a..efc370a148 100644 --- a/tests/lora/test_lora_layers_sd.py +++ b/tests/lora/test_lora_layers_sd.py @@ -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 ) diff --git a/tests/others/test_flashpack.py b/tests/others/test_flashpack.py index 85b7d09fe8..11fd2fc0c4 100644 --- a/tests/others/test_flashpack.py +++ b/tests/others/test_flashpack.py @@ -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) diff --git a/tests/pipelines/kandinsky3/test_kandinsky3.py b/tests/pipelines/kandinsky3/test_kandinsky3.py index 7deae194fb..af58d0b569 100644 --- a/tests/pipelines/kandinsky3/test_kandinsky3.py +++ b/tests/pipelines/kandinsky3/test_kandinsky3.py @@ -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 ) diff --git a/tests/pipelines/kandinsky3/test_kandinsky3_img2img.py b/tests/pipelines/kandinsky3/test_kandinsky3_img2img.py index 469300e677..4d2552ec6c 100644 --- a/tests/pipelines/kandinsky3/test_kandinsky3_img2img.py +++ b/tests/pipelines/kandinsky3/test_kandinsky3_img2img.py @@ -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 ) diff --git a/tests/pipelines/pag/test_pag_sd.py b/tests/pipelines/pag/test_pag_sd.py index 064815d136..24598f23f4 100644 --- a/tests/pipelines/pag/test_pag_sd.py +++ b/tests/pipelines/pag/test_pag_sd.py @@ -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) diff --git a/tests/pipelines/pag/test_pag_sd3_img2img.py b/tests/pipelines/pag/test_pag_sd3_img2img.py index e4146b8780..7bf1af7811 100644 --- a/tests/pipelines/pag/test_pag_sd3_img2img.py +++ b/tests/pipelines/pag/test_pag_sd3_img2img.py @@ -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)"] ) diff --git a/tests/pipelines/pag/test_pag_sd_img2img.py b/tests/pipelines/pag/test_pag_sd_img2img.py index 0b440d5ec9..87c70d3767 100644 --- a/tests/pipelines/pag/test_pag_sd_img2img.py +++ b/tests/pipelines/pag/test_pag_sd_img2img.py @@ -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) diff --git a/tests/pipelines/pag/test_pag_sd_inpaint.py b/tests/pipelines/pag/test_pag_sd_inpaint.py index 754158bbf1..1c16da3cb8 100644 --- a/tests/pipelines/pag/test_pag_sd_inpaint.py +++ b/tests/pipelines/pag/test_pag_sd_inpaint.py @@ -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) diff --git a/tests/pipelines/pag/test_pag_sdxl.py b/tests/pipelines/pag/test_pag_sdxl.py index cca5c61651..a4db47774d 100644 --- a/tests/pipelines/pag/test_pag_sdxl.py +++ b/tests/pipelines/pag/test_pag_sdxl.py @@ -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) diff --git a/tests/pipelines/pag/test_pag_sdxl_img2img.py b/tests/pipelines/pag/test_pag_sdxl_img2img.py index d311500d3c..c3389ecb13 100644 --- a/tests/pipelines/pag/test_pag_sdxl_img2img.py +++ b/tests/pipelines/pag/test_pag_sdxl_img2img.py @@ -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) diff --git a/tests/pipelines/pag/test_pag_sdxl_inpaint.py b/tests/pipelines/pag/test_pag_sdxl_inpaint.py index 00a07582e2..d65dfa229b 100644 --- a/tests/pipelines/pag/test_pag_sdxl_inpaint.py +++ b/tests/pipelines/pag/test_pag_sdxl_inpaint.py @@ -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)