Update app.py
Browse filestorch_dtype=torch.float16,
app.py
CHANGED
@@ -40,7 +40,7 @@ vae = AutoencoderKL.from_pretrained(
|
|
40 |
|
41 |
pipe = StableDiffusionXLFillPipeline.from_pretrained(
|
42 |
"RunDiffusion/Juggernaut-XI-Lightning",
|
43 |
-
|
44 |
vae=vae,
|
45 |
controlnet=model,
|
46 |
).to("cuda")
|
|
|
40 |
|
41 |
pipe = StableDiffusionXLFillPipeline.from_pretrained(
|
42 |
"RunDiffusion/Juggernaut-XI-Lightning",
|
43 |
+
torch_dtype=torch.float16,
|
44 |
vae=vae,
|
45 |
controlnet=model,
|
46 |
).to("cuda")
|