Unverified Commit 7aa6af11 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

[Refactor] Move testing utils out of src (#12238)

* update

* update

* update

* update

* update

* merge main

* Revert "merge main"

This reverts commit 65efbcead58644b31596ed2d714f7cee0e0238d3.
parent 87b800e1
import torch import torch
from diffusers import EulerAncestralDiscreteScheduler from diffusers import EulerAncestralDiscreteScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest from .test_schedulers import SchedulerCommonTest
......
import torch import torch
from diffusers import HeunDiscreteScheduler from diffusers import HeunDiscreteScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest from .test_schedulers import SchedulerCommonTest
......
import torch import torch
from diffusers import KDPM2AncestralDiscreteScheduler from diffusers import KDPM2AncestralDiscreteScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest from .test_schedulers import SchedulerCommonTest
......
import torch import torch
from diffusers import KDPM2DiscreteScheduler from diffusers import KDPM2DiscreteScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest from .test_schedulers import SchedulerCommonTest
......
...@@ -4,8 +4,8 @@ from typing import Dict, List, Tuple ...@@ -4,8 +4,8 @@ from typing import Dict, List, Tuple
import torch import torch
from diffusers import LCMScheduler from diffusers import LCMScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest from .test_schedulers import SchedulerCommonTest
......
import torch import torch
from diffusers import LMSDiscreteScheduler from diffusers import LMSDiscreteScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest from .test_schedulers import SchedulerCommonTest
......
import torch import torch
from diffusers import SASolverScheduler from diffusers import SASolverScheduler
from diffusers.utils.testing_utils import require_torchsde, torch_device
from ..testing_utils import require_torchsde, torch_device
from .test_schedulers import SchedulerCommonTest from .test_schedulers import SchedulerCommonTest
......
...@@ -41,9 +41,9 @@ from diffusers import ( ...@@ -41,9 +41,9 @@ from diffusers import (
from diffusers.configuration_utils import ConfigMixin, register_to_config from diffusers.configuration_utils import ConfigMixin, register_to_config
from diffusers.schedulers.scheduling_utils import SchedulerMixin from diffusers.schedulers.scheduling_utils import SchedulerMixin
from diffusers.utils import logging from diffusers.utils import logging
from diffusers.utils.testing_utils import CaptureLogger, torch_device
from ..others.test_utils import TOKEN, USER, is_staging_test from ..others.test_utils import TOKEN, USER, is_staging_test
from ..testing_utils import CaptureLogger, torch_device
torch.backends.cuda.matmul.allow_tf32 = False torch.backends.cuda.matmul.allow_tf32 = False
......
...@@ -7,7 +7,8 @@ from huggingface_hub import hf_hub_download, snapshot_download ...@@ -7,7 +7,8 @@ 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_utils import _extract_repo_id_and_weights_name
from diffusers.models.attention_processor import AttnProcessor from diffusers.models.attention_processor import AttnProcessor
from diffusers.utils.testing_utils import (
from ..testing_utils import (
numpy_cosine_similarity_distance, numpy_cosine_similarity_distance,
torch_device, torch_device,
) )
......
...@@ -19,7 +19,8 @@ import unittest ...@@ -19,7 +19,8 @@ import unittest
from diffusers import ( from diffusers import (
Lumina2Transformer2DModel, Lumina2Transformer2DModel,
) )
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache, backend_empty_cache,
enable_full_determinism, enable_full_determinism,
require_torch_accelerator, require_torch_accelerator,
......
...@@ -21,7 +21,8 @@ import torch ...@@ -21,7 +21,8 @@ import torch
from diffusers import ( from diffusers import (
AutoencoderDC, AutoencoderDC,
) )
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache, backend_empty_cache,
enable_full_determinism, enable_full_determinism,
load_hf_numpy, load_hf_numpy,
......
...@@ -21,7 +21,8 @@ import torch ...@@ -21,7 +21,8 @@ import torch
from diffusers import ( from diffusers import (
ControlNetModel, ControlNetModel,
) )
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache, backend_empty_cache,
enable_full_determinism, enable_full_determinism,
require_torch_accelerator, require_torch_accelerator,
......
...@@ -19,7 +19,8 @@ import unittest ...@@ -19,7 +19,8 @@ import unittest
from diffusers import ( from diffusers import (
FluxTransformer2DModel, FluxTransformer2DModel,
) )
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache, backend_empty_cache,
enable_full_determinism, enable_full_determinism,
require_torch_accelerator, require_torch_accelerator,
......
...@@ -18,7 +18,8 @@ import unittest ...@@ -18,7 +18,8 @@ import unittest
from diffusers import ( from diffusers import (
MotionAdapter, MotionAdapter,
) )
from diffusers.utils.testing_utils import (
from ..testing_utils import (
enable_full_determinism, enable_full_determinism,
) )
......
...@@ -20,7 +20,8 @@ import torch ...@@ -20,7 +20,8 @@ import torch
from diffusers import StableCascadeUNet from diffusers import StableCascadeUNet
from diffusers.utils import logging from diffusers.utils import logging
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache, backend_empty_cache,
enable_full_determinism, enable_full_determinism,
require_torch_accelerator, require_torch_accelerator,
......
...@@ -21,7 +21,8 @@ import torch ...@@ -21,7 +21,8 @@ import torch
from diffusers import ( from diffusers import (
AutoencoderKL, AutoencoderKL,
) )
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache, backend_empty_cache,
enable_full_determinism, enable_full_determinism,
load_hf_numpy, load_hf_numpy,
......
...@@ -19,7 +19,8 @@ import unittest ...@@ -19,7 +19,8 @@ import unittest
from diffusers import ( from diffusers import (
AutoencoderKLWan, AutoencoderKLWan,
) )
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache, backend_empty_cache,
enable_full_determinism, enable_full_determinism,
require_torch_accelerator, require_torch_accelerator,
......
...@@ -21,7 +21,8 @@ import torch ...@@ -21,7 +21,8 @@ import torch
from diffusers import ( from diffusers import (
WanTransformer3DModel, WanTransformer3DModel,
) )
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache, backend_empty_cache,
enable_full_determinism, enable_full_determinism,
require_big_accelerator, require_big_accelerator,
......
...@@ -4,7 +4,8 @@ import unittest ...@@ -4,7 +4,8 @@ import unittest
from diffusers import ( from diffusers import (
SanaTransformer2DModel, SanaTransformer2DModel,
) )
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache, backend_empty_cache,
enable_full_determinism, enable_full_determinism,
require_torch_accelerator, require_torch_accelerator,
......
...@@ -7,7 +7,8 @@ import torch ...@@ -7,7 +7,8 @@ import torch
from diffusers import ControlNetModel, StableDiffusionControlNetPipeline from diffusers import ControlNetModel, StableDiffusionControlNetPipeline
from diffusers.loaders.single_file_utils import _extract_repo_id_and_weights_name from diffusers.loaders.single_file_utils import _extract_repo_id_and_weights_name
from diffusers.utils import load_image from diffusers.utils import load_image
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache, backend_empty_cache,
enable_full_determinism, enable_full_determinism,
numpy_cosine_similarity_distance, numpy_cosine_similarity_distance,
...@@ -15,7 +16,6 @@ from diffusers.utils.testing_utils import ( ...@@ -15,7 +16,6 @@ from diffusers.utils.testing_utils import (
slow, slow,
torch_device, torch_device,
) )
from .single_file_testing_utils import ( from .single_file_testing_utils import (
SDSingleFileTesterMixin, SDSingleFileTesterMixin,
download_diffusers_config, download_diffusers_config,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment