Spaces:
Running
on
Zero
Running
on
Zero
Update check_app.py
Browse files- 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(
|