Manjushri commited on
Commit
5511722
1 Parent(s): 9a96d60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ torch.cuda.max_memory_allocated(device='cuda')
15
  vae = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", torch_dtype=torch.float16)
16
  torch.cuda.empty_cache()
17
 
18
- def genie (prompt, negative_prompt, scale, steps, seed, upscale):
19
  torch.cuda.max_memory_allocated(device='cuda')
20
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-0.9", torch_dtype=torch.float16, variant="fp16", use_safetensors=True, vae=vae)
21
  pipe = pipe.to(device)
 
15
  vae = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", torch_dtype=torch.float16)
16
  torch.cuda.empty_cache()
17
 
18
+ def genie (prompt, negative_prompt, scale, steps, seed, upscaler):
19
  torch.cuda.max_memory_allocated(device='cuda')
20
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-0.9", torch_dtype=torch.float16, variant="fp16", use_safetensors=True, vae=vae)
21
  pipe = pipe.to(device)