Spaces:

briaai
/

BRIA-2.2-ControlNet-Recoloring

Running on Zero

mokady commited on May 3

Commit

14ac470

verified ·

1 Parent(s): 9e21eaa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,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').to(torch.float16)

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