Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse filesForgot a thing
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 =
|
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()
|