For Ruff check

langgenius · Sep 18, 2024 · a9f024d · a9f024d
1 parent 6f636e1
commit a9f024d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/core/tools/provider/builtin/comfyui/tools/comfyui_stable_diffusion.py b/api/core/tools/provider/builtin/comfyui/tools/comfyui_stable_diffusion.py
@@ -290,7 +290,7 @@ def text2img(
         draw_options["6"]["inputs"]["text"] = prompt
         draw_options["7"]["inputs"]["text"] = negative_prompt
         # if the model is SD3 or FLUX series, the Latent class should be corresponding to SD3 Latent
-        if model_type in (ModelType.SD3.name, ModelType.FLUX.name):
+        if model_type in {ModelType.SD3.name, ModelType.FLUX.name}:
             draw_options["5"]["class_type"] = "EmptySD3LatentImage"
 
         if lora_list: