fixed logic

lib.py

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