Manjushri commited on
Commit
06ace5c
·
verified ·
1 Parent(s): 241d995

Update app.py

Browse files

Forgot another thingy ding

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -278,7 +278,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
278
  sdxl.enable_xformers_memory_efficient_attention()
279
  sdxl = sdxl.to(device)
280
  torch.cuda.empty_cache()
281
- upscaled = sdxl(prompt=Prompt, negative_prompt=negative_prompt, image=image, num_inference_steps=15, guidance_scale=0).images[0]
282
  torch.cuda.empty_cache()
283
  return upscaled
284
  else:
 
278
  sdxl.enable_xformers_memory_efficient_attention()
279
  sdxl = sdxl.to(device)
280
  torch.cuda.empty_cache()
281
+ upscaled = sdxl(prompt=Prompt, negative_prompt=negative_prompt, image=refined, num_inference_steps=15, guidance_scale=0).images[0]
282
  torch.cuda.empty_cache()
283
  return upscaled
284
  else: