TobDeBer commited on
Commit
22e6bfc
·
1 Parent(s): fd54d78
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -7,6 +7,8 @@ import sys
7
  import os
8
  os.environ["BNB_CUDA_VERSION"] = "0" # Forces bitsandbytes to recognize no GPU
9
  os.environ["OMP_NUM_THREADS"] = "1" # Prevents race conditions in custom CPU kernels
 
 
10
 
11
  # Model configuration
12
  if len(sys.argv) > 1 and os.path.exists(sys.argv[1]):
 
7
  import os
8
  os.environ["BNB_CUDA_VERSION"] = "0" # Forces bitsandbytes to recognize no GPU
9
  os.environ["OMP_NUM_THREADS"] = "1" # Prevents race conditions in custom CPU kernels
10
+ os.environ["VECLIB_MAXIMUM_ISA"] = "AVX2"
11
+ os.environ["MKL_DEBUG_CPU_TYPE"] = "5" # Forces MKL to use AVX2
12
 
13
  # Model configuration
14
  if len(sys.argv) > 1 and os.path.exists(sys.argv[1]):