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 (
)
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()
......
......@@ -4,7 +4,7 @@ from diffusers import FlaxAutoencoderKL
from diffusers.utils import is_flax_available
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():
......
......@@ -25,7 +25,7 @@ from diffusers.utils.testing_utils import (
torch_device,
)
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
enable_full_determinism()
......
......@@ -25,7 +25,7 @@ from diffusers.utils.testing_utils import (
torch_device,
)
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
class UNet1DModelTests(ModelTesterMixin, UNetTesterMixin, unittest.TestCase):
......
......@@ -30,7 +30,7 @@ from diffusers.utils.testing_utils import (
torch_device,
)
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
logger = logging.get_logger(__name__)
......
......@@ -48,7 +48,7 @@ from diffusers.utils.testing_utils import (
torch_device,
)
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
logger = logging.get_logger(__name__)
......
......@@ -23,7 +23,7 @@ from diffusers.utils import logging
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 .test_modeling_common import ModelTesterMixin, UNetTesterMixin
from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
enable_full_determinism()
......
......@@ -30,7 +30,7 @@ from diffusers.utils.testing_utils import (
torch_device,
)
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
logger = logging.get_logger(__name__)
......
......@@ -28,7 +28,7 @@ from diffusers.utils.testing_utils import (
torch_device,
)
from .test_modeling_common import ModelTesterMixin, UNetTesterMixin
from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
logger = logging.get_logger(__name__)
......
......@@ -38,7 +38,7 @@ from diffusers.utils.testing_utils import (
torch_device,
)
from ..models.test_models_vae import (
from ..models.autoencoders.test_models_vae import (
get_asym_autoencoder_kl_config,
get_autoencoder_kl_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