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
from diffusers import EulerAncestralDiscreteScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest
......
import torch
from diffusers import HeunDiscreteScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest
......
import torch
from diffusers import KDPM2AncestralDiscreteScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest
......
import torch
from diffusers import KDPM2DiscreteScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest
......
......@@ -4,8 +4,8 @@ from typing import Dict, List, Tuple
import torch
from diffusers import LCMScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest
......
import torch
from diffusers import LMSDiscreteScheduler
from diffusers.utils.testing_utils import torch_device
from ..testing_utils import torch_device
from .test_schedulers import SchedulerCommonTest
......
import torch
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
......
......@@ -41,9 +41,9 @@ from diffusers import (
from diffusers.configuration_utils import ConfigMixin, register_to_config
from diffusers.schedulers.scheduling_utils import SchedulerMixin
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 ..testing_utils import CaptureLogger, torch_device
torch.backends.cuda.matmul.allow_tf32 = False
......
......@@ -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.models.attention_processor import AttnProcessor
from diffusers.utils.testing_utils import (
from ..testing_utils import (
numpy_cosine_similarity_distance,
torch_device,
)
......
......@@ -19,7 +19,8 @@ import unittest
from diffusers import (
Lumina2Transformer2DModel,
)
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache,
enable_full_determinism,
require_torch_accelerator,
......
......@@ -21,7 +21,8 @@ import torch
from diffusers import (
AutoencoderDC,
)
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache,
enable_full_determinism,
load_hf_numpy,
......
......@@ -21,7 +21,8 @@ import torch
from diffusers import (
ControlNetModel,
)
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache,
enable_full_determinism,
require_torch_accelerator,
......
......@@ -19,7 +19,8 @@ import unittest
from diffusers import (
FluxTransformer2DModel,
)
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache,
enable_full_determinism,
require_torch_accelerator,
......
......@@ -18,7 +18,8 @@ import unittest
from diffusers import (
MotionAdapter,
)
from diffusers.utils.testing_utils import (
from ..testing_utils import (
enable_full_determinism,
)
......
......@@ -20,7 +20,8 @@ import torch
from diffusers import StableCascadeUNet
from diffusers.utils import logging
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache,
enable_full_determinism,
require_torch_accelerator,
......
......@@ -21,7 +21,8 @@ import torch
from diffusers import (
AutoencoderKL,
)
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache,
enable_full_determinism,
load_hf_numpy,
......
......@@ -19,7 +19,8 @@ import unittest
from diffusers import (
AutoencoderKLWan,
)
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache,
enable_full_determinism,
require_torch_accelerator,
......
......@@ -21,7 +21,8 @@ import torch
from diffusers import (
WanTransformer3DModel,
)
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache,
enable_full_determinism,
require_big_accelerator,
......
......@@ -4,7 +4,8 @@ import unittest
from diffusers import (
SanaTransformer2DModel,
)
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache,
enable_full_determinism,
require_torch_accelerator,
......
......@@ -7,7 +7,8 @@ import torch
from diffusers import ControlNetModel, StableDiffusionControlNetPipeline
from diffusers.loaders.single_file_utils import _extract_repo_id_and_weights_name
from diffusers.utils import load_image
from diffusers.utils.testing_utils import (
from ..testing_utils import (
backend_empty_cache,
enable_full_determinism,
numpy_cosine_similarity_distance,
......@@ -15,7 +16,6 @@ from diffusers.utils.testing_utils import (
slow,
torch_device,
)
from .single_file_testing_utils import (
SDSingleFileTesterMixin,
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