Manjushri commited on
Commit
fc48bcb
1 Parent(s): bfee72a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def genie (prompt, negative_prompt, height, width, scale, steps, seed, upscaler)
39
  else:
40
  torch.cuda.empty_cache()
41
  torch.cuda.max_memory_allocated(device=device)
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]
 
39
  else:
40
  torch.cuda.empty_cache()
41
  torch.cuda.max_memory_allocated(device=device)
42
+ pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", 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]