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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -231,7 +231,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
231
  animagine.enable_xformers_memory_efficient_attention()
232
  animagine = animagine.to(device)
233
  torch.cuda.empty_cache()
234
- upscaled = refiner(prompt=Prompt, negative_prompt=negative_prompt, image=image, num_inference_steps=15, guidance_scale=0).images[0]
235
  torch.cuda.empty_cache()
236
  return upscaled
237
  else:
 
231
  animagine.enable_xformers_memory_efficient_attention()
232
  animagine = animagine.to(device)
233
  torch.cuda.empty_cache()
234
+ upscaled = animagine(prompt=Prompt, negative_prompt=negative_prompt, image=image, num_inference_steps=15, guidance_scale=0).images[0]
235
  torch.cuda.empty_cache()
236
  return upscaled
237
  else: