Update app.py
Browse files
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,
|
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)
|