kernkraft commited on
Commit
ba74acc
·
verified ·
1 Parent(s): f89a85f

Update app.py

Browse files

torch_dtype=torch.float16,

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ vae = AutoencoderKL.from_pretrained(
40
 
41
  pipe = StableDiffusionXLFillPipeline.from_pretrained(
42
  "RunDiffusion/Juggernaut-XI-Lightning",
43
- dtype=torch.float16,
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")