mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-27 17:22:53 +03:00
[tests] skip dynamo tests when python is 3.12. (#7458)
skip dynamo tests when python is 3.12.
This commit is contained in:
@@ -339,10 +339,15 @@ def deprecate_after_peft_backend(test_case):
|
||||
return unittest.skipUnless(not USE_PEFT_BACKEND, "test skipped in favor of PEFT backend")(test_case)
|
||||
|
||||
|
||||
def get_python_version():
|
||||
sys_info = sys.version_info
|
||||
major, minor = sys_info.major, sys_info.minor
|
||||
return major, minor
|
||||
|
||||
|
||||
def require_python39_or_higher(test_case):
|
||||
def python39_available():
|
||||
sys_info = sys.version_info
|
||||
major, minor = sys_info.major, sys_info.minor
|
||||
major, minor = get_python_version()
|
||||
return major == 3 and minor >= 9
|
||||
|
||||
return unittest.skipUnless(python39_available(), "test requires Python 3.9 or higher")(test_case)
|
||||
|
||||
@@ -34,6 +34,7 @@ from diffusers.training_utils import EMAModel
|
||||
from diffusers.utils import is_xformers_available, logging
|
||||
from diffusers.utils.testing_utils import (
|
||||
CaptureLogger,
|
||||
get_python_version,
|
||||
require_python39_or_higher,
|
||||
require_torch_2,
|
||||
require_torch_accelerator_with_training,
|
||||
@@ -431,6 +432,10 @@ class ModelTesterMixin:
|
||||
|
||||
@require_python39_or_higher
|
||||
@require_torch_2
|
||||
@unittest.skipIf(
|
||||
get_python_version == (3, 12),
|
||||
reason="Torch Dynamo isn't yet supported for Python 3.12.",
|
||||
)
|
||||
def test_from_save_pretrained_dynamo(self):
|
||||
init_dict, _ = self.prepare_init_args_and_inputs_for_common()
|
||||
inputs = [init_dict, self.model_class]
|
||||
|
||||
@@ -35,6 +35,7 @@ from diffusers.pipelines.controlnet.pipeline_controlnet import MultiControlNetMo
|
||||
from diffusers.utils.import_utils import is_xformers_available
|
||||
from diffusers.utils.testing_utils import (
|
||||
enable_full_determinism,
|
||||
get_python_version,
|
||||
load_image,
|
||||
load_numpy,
|
||||
numpy_cosine_similarity_distance,
|
||||
@@ -992,6 +993,10 @@ class ControlNetPipelineSlowTests(unittest.TestCase):
|
||||
|
||||
@require_python39_or_higher
|
||||
@require_torch_2
|
||||
@unittest.skipIf(
|
||||
get_python_version == (3, 12),
|
||||
reason="Torch Dynamo isn't yet supported for Python 3.12.",
|
||||
)
|
||||
def test_stable_diffusion_compile(self):
|
||||
run_test_in_subprocess(test_case=self, target_func=_test_stable_diffusion_compile, inputs=None)
|
||||
|
||||
|
||||
@@ -66,6 +66,7 @@ from diffusers.utils.testing_utils import (
|
||||
CaptureLogger,
|
||||
enable_full_determinism,
|
||||
floats_tensor,
|
||||
get_python_version,
|
||||
get_tests_dir,
|
||||
load_numpy,
|
||||
nightly,
|
||||
@@ -1748,6 +1749,10 @@ class PipelineSlowTests(unittest.TestCase):
|
||||
|
||||
@require_python39_or_higher
|
||||
@require_torch_2
|
||||
@unittest.skipIf(
|
||||
get_python_version == (3, 12),
|
||||
reason="Torch Dynamo isn't yet supported for Python 3.12.",
|
||||
)
|
||||
def test_from_save_pretrained_dynamo(self):
|
||||
run_test_in_subprocess(test_case=self, target_func=_test_from_save_pretrained_dynamo, inputs=None)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user