[TEST] Remove ddim_v3 from test if not model v3

Aedial · Dec 5, 2023 · 62586ad · 62586ad
1 parent 8b99c5a
commit 62586ad
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/api/test_imagegen_samplers.py b/tests/api/test_imagegen_samplers.py
@@ -37,6 +37,10 @@ async def test_samplers(
     api = api_handle.api
     model, sampler = model_sampler
 
+    # ddim_v3 only work with Anime v3
+    if model not in (ImageModel.Anime_v3,) and sampler is ImageSampler.ddim_v3:
+        return
+
     logger = api_handle.logger
     logger.info(f"Testing model {model} with sampler {sampler}")