ford442 commited on
Commit
4bdb15e
·
verified ·
1 Parent(s): b2f53ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -140,7 +140,7 @@ def infer(
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,
146
  guidance_scale=guidance_scale,
@@ -148,7 +148,7 @@ def infer(
148
  width=width,
149
  height=height,
150
  generator=generator
151
- ).images[0]
152
  print('-- got image --')
153
  image_path = f"sd35m_{seed}.png"
154
  sd_image.save(image_path,optimize=False,compress_level=0)
 
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,
146
  guidance_scale=guidance_scale,
 
148
  width=width,
149
  height=height,
150
  generator=generator
151
+ ).images[0]
152
  print('-- got image --')
153
  image_path = f"sd35m_{seed}.png"
154
  sd_image.save(image_path,optimize=False,compress_level=0)