Manjushri commited on
Commit
4444eb4
1 Parent(s): 6446c77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ torch.cuda.empty_cache()
22
  def genie (Prompt, negative_prompt, height, width, scale, steps, seed, upscale):
23
  generator = np.random.seed(0) if seed == 0 else torch.manual_seed(seed)
24
  torch.cuda.empty_cache()
25
- int_image = pipe(prompt=Prompt, negative_prompt=negative_prompt, num_inference_steps=steps, prior_num_inference_steps=steps, prior_guidance_scale=scale, width=width, height=height, output_type="latent").images
26
  torch.cuda.empty_cache()
27
  image = refiner(Prompt, negative_prompt=negative_prompt, image=int_image, denoising_start=.99).images[0]
28
  torch.cuda.empty_cache()
 
22
  def genie (Prompt, negative_prompt, height, width, scale, steps, seed, upscale):
23
  generator = np.random.seed(0) if seed == 0 else torch.manual_seed(seed)
24
  torch.cuda.empty_cache()
25
+ int_image = pipe(prompt=Prompt, negative_prompt=negative_prompt, num_inference_steps=steps, guidance_scale=scale, width=width, height=height, output_type="latent").images
26
  torch.cuda.empty_cache()
27
  image = refiner(Prompt, negative_prompt=negative_prompt, image=int_image, denoising_start=.99).images[0]
28
  torch.cuda.empty_cache()