DamarJati commited on
Commit
76b0dfc
1 Parent(s): 914bc5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -35,6 +35,7 @@ MODEL = os.getenv(
35
 
36
  torch.backends.cudnn.deterministic = True
37
  torch.backends.cudnn.benchmark = False
 
38
 
39
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
40
 
 
35
 
36
  torch.backends.cudnn.deterministic = True
37
  torch.backends.cudnn.benchmark = False
38
+ torch.backends.cuda.matmul.allow_tf32 = True
39
 
40
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
41