Spaces:

briaai
/

BRIA-2.2-ControlNet-Canny

Running on Zero

dvir-bria commited on Feb 8

Commit

c62876e

•

1 Parent(s): cf6c5a9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,8 +17,8 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     controlnet=controlnet,
     torch_dtype=torch.float16,
     device_map=0,
-    low_cpu_mem_usage=True,
-    offload_state_dict=True,
 ).to('cuda')
 pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
 pipe.enable_xformers_memory_efficient_attention()

     controlnet=controlnet,
     torch_dtype=torch.float16,
     device_map=0,
+    # low_cpu_mem_usage=True,
+    # offload_state_dict=True,
 ).to('cuda')
 pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
 pipe.enable_xformers_memory_efficient_attention()