Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|