sivakorn-su commited on
Commit
2954be4
·
1 Parent(s): 723168d

fix config

Browse files
Files changed (1) hide show
  1. config.py +3 -3
config.py CHANGED
@@ -24,9 +24,9 @@ os.environ["TMP"] = "/tmp"
24
  os.environ["XDG_CACHE_HOME"] = "/tmp"
25
  os.environ["HOME"] = "/tmp/home" # Force HOME to writable location
26
  os.environ["PYTHAINLP_DATA_DIR"] = PYTHAI_NLP
27
- os.environ["OMP_NUM_THREADS"] = OMP_NUM_THREADS
28
- os.environ["OPENBLAS_NUM_THREADS"] = OPENBLAS_NUM_THREADS
29
- os.environ["MKL_NUM_THREADS"] = MKL_NUM_THREADS
30
 
31
  torch.set_num_threads(1)
32
  torch.set_num_interop_threads(1)
 
24
  os.environ["XDG_CACHE_HOME"] = "/tmp"
25
  os.environ["HOME"] = "/tmp/home" # Force HOME to writable location
26
  os.environ["PYTHAINLP_DATA_DIR"] = PYTHAI_NLP
27
+ os.environ["OMP_NUM_THREADS"] = "1"
28
+ os.environ["OPENBLAS_NUM_THREADS"] = "1"
29
+ os.environ["MKL_NUM_THREADS"] = "1"
30
 
31
  torch.set_num_threads(1)
32
  torch.set_num_interop_threads(1)