Spaces:

briaai
/

BRIA-2.2-ControlNet-Canny

Running on Zero

dvir-bria commited on Feb 8

Commit

e15e7aa

•

1 Parent(s): ddf2e3c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     "briaai/BRIA-2.0",
     controlnet=controlnet,
     torch_dtype=torch.float16,
-    device_map='auto',
     # low_cpu_mem_usage=True,
     # offload_state_dict=True,
 )#.to('cuda')

     "briaai/BRIA-2.0",
     controlnet=controlnet,
     torch_dtype=torch.float16,
+    device_map='cuda:0',
     # low_cpu_mem_usage=True,
     # offload_state_dict=True,
 )#.to('cuda')