Spaces:
Running
on
Zero
Running
on
Zero
Add new model
Browse files- model/model_manager.py +1 -1
model/model_manager.py
CHANGED
@@ -27,7 +27,7 @@ class ModelManager:
|
|
27 |
@spaces.GPU(duration=120)
|
28 |
def generate_image_ig(self, prompt, model_name):
|
29 |
pipe = self.load_model_pipe(model_name)
|
30 |
-
if 'Stable-cascade' not in
|
31 |
result = pipe(prompt=prompt).images[0]
|
32 |
else:
|
33 |
prior, decoder = pipe
|
|
|
27 |
@spaces.GPU(duration=120)
|
28 |
def generate_image_ig(self, prompt, model_name):
|
29 |
pipe = self.load_model_pipe(model_name)
|
30 |
+
if 'Stable-cascade' not in model_name:
|
31 |
result = pipe(prompt=prompt).images[0]
|
32 |
else:
|
33 |
prior, decoder = pipe
|