Spaces:

fffiloni
/

sdxl-control-loras

Sleeping

fffiloni commited on Sep 21, 2023

Commit

d48fe1d

1 Parent(s): b3f8d5f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,9 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     controlnet=controlnet,
     vae=vae,
-    torch_dtype=torch.float16,
 )
 custom_model = "fffiloni/eugene_jour_general"

     "stabilityai/stable-diffusion-xl-base-1.0",
     controlnet=controlnet,
     vae=vae,
+    torch_dtype=torch.float16,
+    variant="fp16",
+    use_safetensors=True
 )
 custom_model = "fffiloni/eugene_jour_general"