Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
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 =
|
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:
|