Hugging Face's logo

Spaces:
seawolf2357
/
BRIA-2.2-ControlNet-Canny
Runtime error

App Files Community
dvir-bria commited on
Commit
7497bf5
·
1 Parent(s): 63a0180

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
16
  "briaai/BRIA-2.0",
17
  controlnet=controlnet,
18
  torch_dtype=torch.float16,
19
- )
20
  pipe.enable_xformers_memory_efficient_attention()
21
 
22
  low_threshold = 100
 
16
  "briaai/BRIA-2.0",
17
  controlnet=controlnet,
18
  torch_dtype=torch.float16,
19
+ ).cuda()
20
  pipe.enable_xformers_memory_efficient_attention()
21
 
22
  low_threshold = 100