"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "dfcce3ca6ed4cd68246bdf7ee98f1e67b4d252c4"
Unverified Commit ec9840a5 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Refactor] harmonize the module structure for models in tests (#6738)



* harmonize the module structure for models in tests

* make the folders modules.

---------
Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent 093a03a1
...@@ -46,7 +46,7 @@ from diffusers.utils.testing_utils import ( ...@@ -46,7 +46,7 @@ from diffusers.utils.testing_utils import (
) )
from diffusers.utils.torch_utils import randn_tensor from diffusers.utils.torch_utils import randn_tensor
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
enable_full_determinism() enable_full_determinism()
......
...@@ -4,7 +4,7 @@ from diffusers import FlaxAutoencoderKL ...@@ -4,7 +4,7 @@ from diffusers import FlaxAutoencoderKL
from diffusers.utils import is_flax_available from diffusers.utils import is_flax_available
from diffusers.utils.testing_utils import require_flax from diffusers.utils.testing_utils import require_flax
from .test_modeling_common_flax import FlaxModelTesterMixin from ..test_modeling_common_flax import FlaxModelTesterMixin
if is_flax_available(): if is_flax_available():
......
...@@ -25,7 +25,7 @@ from diffusers.utils.testing_utils import ( ...@@ -25,7 +25,7 @@ from diffusers.utils.testing_utils import (
torch_device, torch_device,
) )
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
enable_full_determinism() enable_full_determinism()
......
...@@ -25,7 +25,7 @@ from diffusers.utils.testing_utils import ( ...@@ -25,7 +25,7 @@ from diffusers.utils.testing_utils import (
torch_device, torch_device,
) )
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
class UNet1DModelTests(ModelTesterMixin, UNetTesterMixin, unittest.TestCase): class UNet1DModelTests(ModelTesterMixin, UNetTesterMixin, unittest.TestCase):
......
...@@ -30,7 +30,7 @@ from diffusers.utils.testing_utils import ( ...@@ -30,7 +30,7 @@ from diffusers.utils.testing_utils import (
torch_device, torch_device,
) )
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
......
...@@ -48,7 +48,7 @@ from diffusers.utils.testing_utils import ( ...@@ -48,7 +48,7 @@ from diffusers.utils.testing_utils import (
torch_device, torch_device,
) )
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
......
...@@ -23,7 +23,7 @@ from diffusers.utils import logging ...@@ -23,7 +23,7 @@ from diffusers.utils import logging
from diffusers.utils.import_utils import is_xformers_available from diffusers.utils.import_utils import is_xformers_available
from diffusers.utils.testing_utils import enable_full_determinism, floats_tensor, skip_mps, torch_device from diffusers.utils.testing_utils import enable_full_determinism, floats_tensor, skip_mps, torch_device
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
enable_full_determinism() enable_full_determinism()
......
...@@ -30,7 +30,7 @@ from diffusers.utils.testing_utils import ( ...@@ -30,7 +30,7 @@ from diffusers.utils.testing_utils import (
torch_device, torch_device,
) )
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
......
...@@ -28,7 +28,7 @@ from diffusers.utils.testing_utils import ( ...@@ -28,7 +28,7 @@ from diffusers.utils.testing_utils import (
torch_device, torch_device,
) )
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
......
...@@ -38,7 +38,7 @@ from diffusers.utils.testing_utils import ( ...@@ -38,7 +38,7 @@ from diffusers.utils.testing_utils import (
torch_device, torch_device,
) )
from ..models.test_models_vae import ( from ..models.autoencoders.test_models_vae import (
get_asym_autoencoder_kl_config, get_asym_autoencoder_kl_config,
get_autoencoder_kl_config, get_autoencoder_kl_config,
get_autoencoder_tiny_config, get_autoencoder_tiny_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