Spaces:

briaai
/

BRIA-2.2-ControlNet-Canny

Running on Zero

mokady commited on 28 days ago

Commit

adacf46

verified ·

1 Parent(s): cc497c8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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