ostapagon commited on
Commit
5f75751
·
1 Parent(s): bc27ec5

Lower tf memory footprint

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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()