lib.py

@@ -200,9 +200,10 @@ def generate_image(
         # Conditionally set model if node and param are provided
         if model_node and model_param:
             if user_config["comfyui"].get("FLUX"):
-                valid_models = user_config["comfyui:flux"]["models"].split(",")
+                available_model_list = user_config["comfyui"]["confyui:flux"]["models"].split(",")
             else:
                 available_model_list = user_config["comfyui"]["models"].split(",")
+
             valid_models = list(
                 set(get_available_models()) & set(available_model_list)
             )