erikbeltran commited on
Commit
7645e5e
·
verified ·
1 Parent(s): 4809f80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def generate_image(prompt, ckpt):
60
  pipe.unet.load_state_dict(load_file(hf_hub_download(repo, checkpoint), device="cuda"))
61
  loaded = num_inference_steps
62
 
63
- results = pipe(prompt, num_inference_steps=num_inference_steps, width=512, height=256, guidance_scale=0)
64
 
65
  if SAFETY_CHECKER:
66
  images, has_nsfw_concepts = check_nsfw_images(results.images)
 
60
  pipe.unet.load_state_dict(load_file(hf_hub_download(repo, checkpoint), device="cuda"))
61
  loaded = num_inference_steps
62
 
63
+ results = pipe(prompt, num_inference_steps=num_inference_steps, width=1024, height=512, guidance_scale=0)
64
 
65
  if SAFETY_CHECKER:
66
  images, has_nsfw_concepts = check_nsfw_images(results.images)