1
0
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:
Sayak Paul
2024-03-26 08:39:48 +05:30
committed by GitHub
parent 0dd0528851
commit 484c8ef399
4 changed files with 22 additions and 2 deletions

View File

@@ -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)

View File

@@ -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]

View File

@@ -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)

View File

@@ -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)