diff --git a/tests/single_file/single_file_testing_utils.py b/tests/single_file/single_file_testing_utils.py index 4e1713c9ce..10c229ba5e 100644 --- a/tests/single_file/single_file_testing_utils.py +++ b/tests/single_file/single_file_testing_utils.py @@ -5,7 +5,7 @@ import requests import torch from huggingface_hub import hf_hub_download, snapshot_download -from diffusers.loaders.single_file_utils import _extract_repo_id_and_weights_name +from diffusers.loaders.single_file.single_file_utils import _extract_repo_id_and_weights_name from diffusers.models.attention_processor import AttnProcessor from diffusers.utils.testing_utils import ( numpy_cosine_similarity_distance, diff --git a/tests/single_file/test_model_vae_single_file.py b/tests/single_file/test_model_vae_single_file.py index bba1726ae3..350a433cf6 100644 --- a/tests/single_file/test_model_vae_single_file.py +++ b/tests/single_file/test_model_vae_single_file.py @@ -18,9 +18,7 @@ import unittest import torch -from diffusers import ( - AutoencoderKL, -) +from diffusers import AutoencoderKL from diffusers.utils.testing_utils import ( backend_empty_cache, enable_full_determinism, diff --git a/tests/single_file/test_model_wan_autoencoder_single_file.py b/tests/single_file/test_model_wan_autoencoder_single_file.py index 7f0e1c1a4b..83acbdc565 100644 --- a/tests/single_file/test_model_wan_autoencoder_single_file.py +++ b/tests/single_file/test_model_wan_autoencoder_single_file.py @@ -16,9 +16,7 @@ import gc import unittest -from diffusers import ( - AutoencoderKLWan, -) +from diffusers import AutoencoderKLWan from diffusers.utils.testing_utils import ( backend_empty_cache, enable_full_determinism, diff --git a/tests/single_file/test_model_wan_transformer3d_single_file.py b/tests/single_file/test_model_wan_transformer3d_single_file.py index 36f0919cac..8a20135a8a 100644 --- a/tests/single_file/test_model_wan_transformer3d_single_file.py +++ b/tests/single_file/test_model_wan_transformer3d_single_file.py @@ -18,9 +18,7 @@ import unittest import torch -from diffusers import ( - WanTransformer3DModel, -) +from diffusers import WanTransformer3DModel from diffusers.utils.testing_utils import ( backend_empty_cache, enable_full_determinism, diff --git a/tests/single_file/test_sana_transformer.py b/tests/single_file/test_sana_transformer.py index 802ca37abf..061a5698d3 100644 --- a/tests/single_file/test_sana_transformer.py +++ b/tests/single_file/test_sana_transformer.py @@ -3,9 +3,7 @@ import unittest import torch -from diffusers import ( - SanaTransformer2DModel, -) +from diffusers import SanaTransformer2DModel from diffusers.utils.testing_utils import ( backend_empty_cache, enable_full_determinism, diff --git a/tests/single_file/test_stable_diffusion_controlnet_img2img_single_file.py b/tests/single_file/test_stable_diffusion_controlnet_img2img_single_file.py index 7589b48028..2997224115 100644 --- a/tests/single_file/test_stable_diffusion_controlnet_img2img_single_file.py +++ b/tests/single_file/test_stable_diffusion_controlnet_img2img_single_file.py @@ -5,7 +5,7 @@ import unittest import torch from diffusers import ControlNetModel, StableDiffusionControlNetPipeline -from diffusers.loaders.single_file_utils import _extract_repo_id_and_weights_name +from diffusers.loaders.single_file.single_file_utils import _extract_repo_id_and_weights_name from diffusers.utils import load_image from diffusers.utils.testing_utils import ( backend_empty_cache, diff --git a/tests/single_file/test_stable_diffusion_controlnet_inpaint_single_file.py b/tests/single_file/test_stable_diffusion_controlnet_inpaint_single_file.py index 1555831db6..166365a9a0 100644 --- a/tests/single_file/test_stable_diffusion_controlnet_inpaint_single_file.py +++ b/tests/single_file/test_stable_diffusion_controlnet_inpaint_single_file.py @@ -5,7 +5,7 @@ import unittest import torch from diffusers import ControlNetModel, StableDiffusionControlNetInpaintPipeline -from diffusers.loaders.single_file_utils import _extract_repo_id_and_weights_name +from diffusers.loaders.single_file.single_file_utils import _extract_repo_id_and_weights_name from diffusers.utils import load_image from diffusers.utils.testing_utils import ( backend_empty_cache, diff --git a/tests/single_file/test_stable_diffusion_controlnet_single_file.py b/tests/single_file/test_stable_diffusion_controlnet_single_file.py index 2c1e414e5e..35ce5daac2 100644 --- a/tests/single_file/test_stable_diffusion_controlnet_single_file.py +++ b/tests/single_file/test_stable_diffusion_controlnet_single_file.py @@ -5,7 +5,7 @@ import unittest import torch from diffusers import ControlNetModel, StableDiffusionControlNetPipeline -from diffusers.loaders.single_file_utils import _extract_repo_id_and_weights_name +from diffusers.loaders.single_file.single_file_utils import _extract_repo_id_and_weights_name from diffusers.utils import load_image from diffusers.utils.testing_utils import ( backend_empty_cache, diff --git a/tests/single_file/test_stable_diffusion_single_file.py b/tests/single_file/test_stable_diffusion_single_file.py index 78baeb9492..338842e8d0 100644 --- a/tests/single_file/test_stable_diffusion_single_file.py +++ b/tests/single_file/test_stable_diffusion_single_file.py @@ -5,7 +5,7 @@ import unittest import torch from diffusers import EulerDiscreteScheduler, StableDiffusionInstructPix2PixPipeline, StableDiffusionPipeline -from diffusers.loaders.single_file_utils import _extract_repo_id_and_weights_name +from diffusers.loaders.single_file.single_file_utils import _extract_repo_id_and_weights_name from diffusers.utils import load_image from diffusers.utils.testing_utils import ( backend_empty_cache, diff --git a/tests/single_file/test_stable_diffusion_xl_adapter_single_file.py b/tests/single_file/test_stable_diffusion_xl_adapter_single_file.py index fb5f8725b8..06a4f0bf5b 100644 --- a/tests/single_file/test_stable_diffusion_xl_adapter_single_file.py +++ b/tests/single_file/test_stable_diffusion_xl_adapter_single_file.py @@ -8,7 +8,7 @@ from diffusers import ( StableDiffusionXLAdapterPipeline, T2IAdapter, ) -from diffusers.loaders.single_file_utils import _extract_repo_id_and_weights_name +from diffusers.loaders.single_file.single_file_utils import _extract_repo_id_and_weights_name from diffusers.utils import load_image from diffusers.utils.testing_utils import ( backend_empty_cache, diff --git a/tests/single_file/test_stable_diffusion_xl_controlnet_single_file.py b/tests/single_file/test_stable_diffusion_xl_controlnet_single_file.py index 6d8c4369e1..9ea2fcc297 100644 --- a/tests/single_file/test_stable_diffusion_xl_controlnet_single_file.py +++ b/tests/single_file/test_stable_diffusion_xl_controlnet_single_file.py @@ -5,7 +5,7 @@ import unittest import torch from diffusers import ControlNetModel, StableDiffusionXLControlNetPipeline -from diffusers.loaders.single_file_utils import _extract_repo_id_and_weights_name +from diffusers.loaders.single_file.single_file_utils import _extract_repo_id_and_weights_name from diffusers.utils import load_image from diffusers.utils.testing_utils import ( backend_empty_cache,