Unverified Commit 484c8ef3 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[tests] skip dynamo tests when python is 3.12. (#7458)

skip dynamo tests when python is 3.12.
parent 0dd05288
...@@ -339,10 +339,15 @@ def deprecate_after_peft_backend(test_case): ...@@ -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) 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 require_python39_or_higher(test_case):
def python39_available(): def python39_available():
sys_info = sys.version_info major, minor = get_python_version()
major, minor = sys_info.major, sys_info.minor
return major == 3 and minor >= 9 return major == 3 and minor >= 9
return unittest.skipUnless(python39_available(), "test requires Python 3.9 or higher")(test_case) return unittest.skipUnless(python39_available(), "test requires Python 3.9 or higher")(test_case)
......
...@@ -34,6 +34,7 @@ from diffusers.training_utils import EMAModel ...@@ -34,6 +34,7 @@ from diffusers.training_utils import EMAModel
from diffusers.utils import is_xformers_available, logging from diffusers.utils import is_xformers_available, logging
from diffusers.utils.testing_utils import ( from diffusers.utils.testing_utils import (
CaptureLogger, CaptureLogger,
get_python_version,
require_python39_or_higher, require_python39_or_higher,
require_torch_2, require_torch_2,
require_torch_accelerator_with_training, require_torch_accelerator_with_training,
...@@ -431,6 +432,10 @@ class ModelTesterMixin: ...@@ -431,6 +432,10 @@ class ModelTesterMixin:
@require_python39_or_higher @require_python39_or_higher
@require_torch_2 @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): def test_from_save_pretrained_dynamo(self):
init_dict, _ = self.prepare_init_args_and_inputs_for_common() init_dict, _ = self.prepare_init_args_and_inputs_for_common()
inputs = [init_dict, self.model_class] inputs = [init_dict, self.model_class]
......
...@@ -35,6 +35,7 @@ from diffusers.pipelines.controlnet.pipeline_controlnet import MultiControlNetMo ...@@ -35,6 +35,7 @@ from diffusers.pipelines.controlnet.pipeline_controlnet import MultiControlNetMo
from diffusers.utils.import_utils import is_xformers_available from diffusers.utils.import_utils import is_xformers_available
from diffusers.utils.testing_utils import ( from diffusers.utils.testing_utils import (
enable_full_determinism, enable_full_determinism,
get_python_version,
load_image, load_image,
load_numpy, load_numpy,
numpy_cosine_similarity_distance, numpy_cosine_similarity_distance,
...@@ -992,6 +993,10 @@ class ControlNetPipelineSlowTests(unittest.TestCase): ...@@ -992,6 +993,10 @@ class ControlNetPipelineSlowTests(unittest.TestCase):
@require_python39_or_higher @require_python39_or_higher
@require_torch_2 @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): def test_stable_diffusion_compile(self):
run_test_in_subprocess(test_case=self, target_func=_test_stable_diffusion_compile, inputs=None) 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 ( ...@@ -66,6 +66,7 @@ from diffusers.utils.testing_utils import (
CaptureLogger, CaptureLogger,
enable_full_determinism, enable_full_determinism,
floats_tensor, floats_tensor,
get_python_version,
get_tests_dir, get_tests_dir,
load_numpy, load_numpy,
nightly, nightly,
...@@ -1748,6 +1749,10 @@ class PipelineSlowTests(unittest.TestCase): ...@@ -1748,6 +1749,10 @@ class PipelineSlowTests(unittest.TestCase):
@require_python39_or_higher @require_python39_or_higher
@require_torch_2 @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): def test_from_save_pretrained_dynamo(self):
run_test_in_subprocess(test_case=self, target_func=_test_from_save_pretrained_dynamo, inputs=None) run_test_in_subprocess(test_case=self, target_func=_test_from_save_pretrained_dynamo, inputs=None)
......
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