Spaces:

briaai
/

BRIA-2.2-ControlNet-Canny

Running on Zero

dvir-bria commited on Dec 27, 2023

Commit

5cef634

1 Parent(s): d2a2088

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,6 @@ controlnet = ControlNetModel.from_pretrained(
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     "briaai/BRIA-2.0",
     controlnet=controlnet,
-    vae=vae,
     torch_dtype=torch.float16,
 )
 pipe.enable_model_cpu_offload()

 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     "briaai/BRIA-2.0",
     controlnet=controlnet,
     torch_dtype=torch.float16,
 )
 pipe.enable_model_cpu_offload()