From 62586adc05b77d2790b4821beddf977ebc54403f Mon Sep 17 00:00:00 2001
From: Aedial <aedial.dev@gmail.com>
Date: Tue, 5 Dec 2023 01:26:23 +0100
Subject: [PATCH] [TEST] Remove ddim_v3 from test if not model v3

---
 tests/api/test_imagegen_samplers.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/api/test_imagegen_samplers.py b/tests/api/test_imagegen_samplers.py
index 0240354..caa364c 100644
--- 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}")