Spaces:
Sleeping
Sleeping
pin avx2
Browse files
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]):
|