Manjushri commited on
Commit
8cee0dc
·
verified ·
1 Parent(s): 15435c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)