Manjushri commited on
Commit
b08ecda
·
1 Parent(s): 01807fb

Update app.py

Browse files

Added x2 upscaler. Fixed pipe

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ pipe = pipe.to(device)
14
  def genie (prompt, scale, steps, Seed):
15
  generator = torch.Generator(device=device).manual_seed(Seed)
16
  #images = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
17
- low_res_latents = pipeline(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator, output_type="latent").images
18
  upscaled_image = upscaler(prompt=prompt, image=low_res_latents, num_inference_steps=20, guidance_scale=0, generator=generator).images[0]
19
  return upscaled_image
20
 
 
14
  def genie (prompt, scale, steps, Seed):
15
  generator = torch.Generator(device=device).manual_seed(Seed)
16
  #images = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
17
+ low_res_latents = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator, output_type="latent").images
18
  upscaled_image = upscaler(prompt=prompt, image=low_res_latents, num_inference_steps=20, guidance_scale=0, generator=generator).images[0]
19
  return upscaled_image
20