remiai3 commited on
Commit
61df085
·
verified ·
1 Parent(s): 84e9994

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -92,7 +92,7 @@ def generate():
92
  pipe.to(torch.device("cpu"))
93
 
94
  images = []
95
- num_inference_steps = 20 if model_id == 'ssd-1b' else 30
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,