husseinelsaadi commited on
Commit
1de8504
·
1 Parent(s): 2a8ccfd
Files changed (1) hide show
  1. backend/services/resume_parser.py +6 -3
backend/services/resume_parser.py CHANGED
@@ -8,9 +8,12 @@ import torch
8
  from typing import List
9
  from transformers import AutoModelForCausalLM, AutoTokenizer
10
 
11
- # ===============================
12
- # Load DeepSeek Janus-Pro-7B Model
13
- # ===============================
 
 
 
14
  MODEL_ID = "tiiuae/falcon-7b-instruct"
15
 
16
 
 
8
  from typing import List
9
  from transformers import AutoModelForCausalLM, AutoTokenizer
10
 
11
+ os.environ["OMP_NUM_THREADS"] = "1"
12
+ os.environ["OPENBLAS_NUM_THREADS"] = "1"
13
+ os.environ["MKL_NUM_THREADS"] = "1"
14
+ os.environ["NUMEXPR_NUM_THREADS"] = "1"
15
+ os.environ["VECLIB_MAXIMUM_THREADS"] = "1"
16
+
17
  MODEL_ID = "tiiuae/falcon-7b-instruct"
18
 
19