Manjushri commited on
Commit
c55678d
1 Parent(s): 1b1ec40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def genie (prompt, negative_prompt, height, width, scale, steps, seed, upscaler)
42
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
43
  pipe = pipe.to(device)
44
  pipe.enable_xformers_memory_efficient_attention()
45
- image = pipe(prompt=prompt, image=int_image).images[0]
46
  torch.cuda.empty_cache()
47
  return (image, image)
48
 
 
42
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
43
  pipe = pipe.to(device)
44
  pipe.enable_xformers_memory_efficient_attention()
45
+ image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=int_image).images[0]
46
  torch.cuda.empty_cache()
47
  return (image, image)
48