NeeravS commited on
Commit
5e8743b
·
verified ·
1 Parent(s): dc19982

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -8,6 +8,8 @@ import json
8
  import spaces
9
  import torch
10
 
 
 
11
  torch.use_deterministic_algorithms(True)
12
  torch.backends.cudnn.deterministic = True
13
  torch.backends.cudnn.benchmark = False
 
8
  import spaces
9
  import torch
10
 
11
+ torch.set_num_threads(1)
12
+ torch.set_num_interop_threads(1)
13
  torch.use_deterministic_algorithms(True)
14
  torch.backends.cudnn.deterministic = True
15
  torch.backends.cudnn.benchmark = False