m7n commited on
Commit
ff632cb
·
1 Parent(s): a6bf88c

set Numba threads to 16 and print updated thread count

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -20,6 +20,8 @@ os.environ["TRANSFORMERS_CACHE"] = os.path.abspath("./transformers_cache")
20
  import numba
21
  print(numba.config)
22
  print("Numba threads:", numba.get_num_threads())
 
 
23
 
24
  from pathlib import Path
25
  from datetime import datetime
 
20
  import numba
21
  print(numba.config)
22
  print("Numba threads:", numba.get_num_threads())
23
+ numba.set_num_threads(16)
24
+ print("Updated Numba threads:", numba.get_num_threads())
25
 
26
  from pathlib import Path
27
  from datetime import datetime