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,
|