Manjushri commited on
Commit
bc862a1
·
verified ·
1 Parent(s): 2cf2468

Update app.py

Browse files

Forgot a thing

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
30
  if upscale == "Yes":
31
  refiner = DiffusionPipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16, use_safetensors=True)
32
  refiner.enable_xformers_memory_efficient_attention()
33
- refiner = upscaler.to(device)
34
  torch.cuda.empty_cache()
35
  upscaled = refiner(prompt=Prompt, negative_prompt=negative_prompt, image=image, num_inference_steps=15, guidance_scale=0).images[0]
36
  torch.cuda.empty_cache()
 
30
  if upscale == "Yes":
31
  refiner = DiffusionPipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16, use_safetensors=True)
32
  refiner.enable_xformers_memory_efficient_attention()
33
+ refiner = refiner.to(device)
34
  torch.cuda.empty_cache()
35
  upscaled = refiner(prompt=Prompt, negative_prompt=negative_prompt, image=image, num_inference_steps=15, guidance_scale=0).images[0]
36
  torch.cuda.empty_cache()