Manjushri commited on
Commit
25ad1b8
·
1 Parent(s): d41ad4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def genie (Prompt, negative_prompt, height, width, scale, steps, Seed, upscale):
19
  generator = torch.Generator(device=device).manual_seed(Seed)
20
  if upscale == "Yes":
21
  image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
22
- upscaled = upscaler(Prompt, negative_prompt=negative_prompt, image=image, num_inference_steps=25, guidance_scale=0, generator=generator).images[0]
23
  return (image, upscaled)
24
  else:
25
  image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
 
19
  generator = torch.Generator(device=device).manual_seed(Seed)
20
  if upscale == "Yes":
21
  image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
22
+ upscaled = upscaler(Prompt, negative_prompt=negative_prompt, image=image, num_inference_steps=5, guidance_scale=0, generator=generator).images[0]
23
  return (image, upscaled)
24
  else:
25
  image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]