ford442 commited on
Commit
05a48aa
Β·
1 Parent(s): 7db43ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -181,8 +181,7 @@ def generate(
181
  options["use_resolution_binning"] = True
182
 
183
  images = []
184
- with torch.no_grad():
185
- for i in range(0, num_images, BATCH_SIZE):
186
  batch_options = options.copy()
187
  batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
188
  if "negative_prompt" in batch_options:
 
181
  options["use_resolution_binning"] = True
182
 
183
  images = []
184
+ for i in range(0, num_images, BATCH_SIZE):
 
185
  batch_options = options.copy()
186
  batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
187
  if "negative_prompt" in batch_options: