RageshAntony commited on
Commit
e06b289
·
verified ·
1 Parent(s): 7fe5aae

Update check_app.py

Browse files
Files changed (1) hide show
  1. check_app.py +1 -1
check_app.py CHANGED
@@ -181,7 +181,7 @@ def create_pipeline_logic(prompt_text, model_name, negative_prompt="", seed=42,
181
  config = MODEL_CONFIGS[model_name]
182
  pipe_class = config["pipeline_class"]
183
  global pipe
184
- if pipe["pipeline_class"] != pipe_class:
185
  print(f"NO PIPE. LOADING NEW {model_name}")
186
  pipe["pipeline_class"] = pipe_class
187
  b_pipe = AutoPipelineForText2Image.from_pretrained(
 
181
  config = MODEL_CONFIGS[model_name]
182
  pipe_class = config["pipeline_class"]
183
  global pipe
184
+ if "pipeline_class" not in pipe or pipe["pipeline_class"] != pipe_class:
185
  print(f"NO PIPE. LOADING NEW {model_name}")
186
  pipe["pipeline_class"] = pipe_class
187
  b_pipe = AutoPipelineForText2Image.from_pretrained(