Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -39,6 +39,8 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
|
|
39 |
return image
|
40 |
else:
|
41 |
if upscale == "Yes":
|
|
|
|
|
42 |
upscaler = DiffusionPipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16, use_safetensors=True)
|
43 |
upscaler.enable_xformers_memory_efficient_attention()
|
44 |
upscaler = upscaler.to(device)
|
|
|
39 |
return image
|
40 |
else:
|
41 |
if upscale == "Yes":
|
42 |
+
image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale).images[0]
|
43 |
+
|
44 |
upscaler = DiffusionPipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16, use_safetensors=True)
|
45 |
upscaler.enable_xformers_memory_efficient_attention()
|
46 |
upscaler = upscaler.to(device)
|