fix base_model bug

edenartlab · Jan 5, 2025 · 3b1e6b5 · 3b1e6b5
1 parent 8fa9e97
commit 3b1e6b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfyui.py b/comfyui.py
@@ -719,7 +719,7 @@ def validate_url(url):
 
             # if there's a lora, replace mentions with embedding name
             if key == "prompt":
-                if "flux" in base_model:
+                if base_model == "flux-dev":
                     for lora_key in ["lora", "lora2"]:
                         if args.get(f"use_{lora_key}", False):
                             lora_strength = args.get(f"{lora_key}_strength", 0.7)