Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -92,7 +92,7 @@ def generate():
|
|
92 |
pipe.to(torch.device("cpu"))
|
93 |
|
94 |
images = []
|
95 |
-
num_inference_steps =
|
96 |
for _ in range(num_images):
|
97 |
image = pipe(
|
98 |
prompt=prompt,
|
|
|
92 |
pipe.to(torch.device("cpu"))
|
93 |
|
94 |
images = []
|
95 |
+
num_inference_steps = 30 if model_id == 'ssd-1b' else 40
|
96 |
for _ in range(num_images):
|
97 |
image = pipe(
|
98 |
prompt=prompt,
|