ford442 commited on
Commit
b2f53ee
·
verified ·
1 Parent(s): 11d8348

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ def infer(
139
  print('-- filtered prompt --')
140
  print(enhanced_prompt)
141
  print('-- generating image --')
142
- with torch.no_grad():
143
  sd_image = pipe(
144
  prompt=enhanced_prompt, # This conversion is fine
145
  negative_prompt=negative_prompt,
 
139
  print('-- filtered prompt --')
140
  print(enhanced_prompt)
141
  print('-- generating image --')
142
+ #with torch.no_grad():
143
  sd_image = pipe(
144
  prompt=enhanced_prompt, # This conversion is fine
145
  negative_prompt=negative_prompt,