Spaces:
Running
on
Zero
Running
on
Zero
Lower tf memory footprint
Browse files
app.py
CHANGED
@@ -8,6 +8,8 @@ from mast3r.demo import get_args_parser
|
|
8 |
from mast3r_demo import mast3r_demo_tab
|
9 |
from gs_demo import gs_demo_tab
|
10 |
|
|
|
|
|
11 |
if __name__ == '__main__':
|
12 |
# parser = get_args_parser()
|
13 |
# args = parser.parse_args()
|
|
|
8 |
from mast3r_demo import mast3r_demo_tab
|
9 |
from gs_demo import gs_demo_tab
|
10 |
|
11 |
+
torch.backends.cuda.matmul.allow_tf32 = True
|
12 |
+
|
13 |
if __name__ == '__main__':
|
14 |
# parser = get_args_parser()
|
15 |
# args = parser.parse_args()
|