vinay0123 commited on
Commit
25f00f3
·
verified ·
1 Parent(s): 7353b52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ import io
23
 
24
  # Set PyTorch to use all available CPU threads
25
  torch.set_num_threads(os.cpu_count())
26
- torch.set_num_interop_threads(os.cpu_count())
27
 
28
  # Enable PyTorch JIT for better performance
29
  torch.jit.enable_onednn_fusion(True)
 
23
 
24
  # Set PyTorch to use all available CPU threads
25
  torch.set_num_threads(os.cpu_count())
26
+ # torch.set_num_interop_threads(os.cpu_count())
27
 
28
  # Enable PyTorch JIT for better performance
29
  torch.jit.enable_onednn_fusion(True)