Hugging Face's logo

Spaces:
briaai
/
BRIA-2.2-ControlNet-Canny
Running on Zero

App Files Community
dvir-bria commited on
Commit
37e1426
•
1 Parent(s): 24591c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ pipe.scheduler = EulerAncestralDiscreteScheduler(
67
  # pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
68
  # pipe.enable_xformers_memory_efficient_attention()
69
  pipe.force_zeros_for_empty_prompt = False
70
- pipe.to('cuda')
71
  low_threshold = 100
72
  high_threshold = 200
73
 
 
67
  # pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
68
  # pipe.enable_xformers_memory_efficient_attention()
69
  pipe.force_zeros_for_empty_prompt = False
70
+ pipe.to('cuda').to(torch.float16)
71
  low_threshold = 100
72
  high_threshold = 200
73