m97j commited on
Commit
b22b0e4
ยท
1 Parent(s): 9d009ba

Initial codes commit

Browse files
Files changed (2) hide show
  1. models/initializer.py +4 -2
  2. requirements.txt +1 -0
models/initializer.py CHANGED
@@ -9,7 +9,8 @@ def initialize_models(app: FastAPI):
9
  # Embedder
10
  embedder_tokenizer = AutoTokenizer.from_pretrained(
11
  HF_MODEL_REPO_ID,
12
- subfolder=EMBED_DIR # ํ† ํฌ๋‚˜์ด์ € ๊ด€๋ จ ํŒŒ์ผ์ด embedder/ ์•ˆ์— ์žˆ์œผ๋ฏ€๋กœ ์ง€์ •
 
13
  )
14
  embedder_model_path = hf_hub_download(
15
  repo_id=HF_MODEL_REPO_ID,
@@ -21,7 +22,8 @@ def initialize_models(app: FastAPI):
21
  # Reranker
22
  reranker_tokenizer = AutoTokenizer.from_pretrained(
23
  HF_MODEL_REPO_ID,
24
- subfolder=RERANK_DIR # ํ† ํฌ๋‚˜์ด์ € ๊ด€๋ จ ํŒŒ์ผ์ด reranker/ ์•ˆ์— ์žˆ์œผ๋ฏ€๋กœ ์ง€์ •
 
25
  )
26
  reranker_model_path = hf_hub_download(
27
  repo_id=HF_MODEL_REPO_ID,
 
9
  # Embedder
10
  embedder_tokenizer = AutoTokenizer.from_pretrained(
11
  HF_MODEL_REPO_ID,
12
+ subfolder=EMBED_DIR, # ํ† ํฌ๋‚˜์ด์ € ๊ด€๋ จ ํŒŒ์ผ์ด embedder/ ์•ˆ์— ์žˆ์œผ๋ฏ€๋กœ ์ง€์ •
13
+ trust_remote_code=True
14
  )
15
  embedder_model_path = hf_hub_download(
16
  repo_id=HF_MODEL_REPO_ID,
 
22
  # Reranker
23
  reranker_tokenizer = AutoTokenizer.from_pretrained(
24
  HF_MODEL_REPO_ID,
25
+ subfolder=RERANK_DIR, # ํ† ํฌ๋‚˜์ด์ € ๊ด€๋ จ ํŒŒ์ผ์ด reranker/ ์•ˆ์— ์žˆ์œผ๋ฏ€๋กœ ์ง€์ •
26
+ trust_remote_code=True
27
  )
28
  reranker_model_path = hf_hub_download(
29
  repo_id=HF_MODEL_REPO_ID,
requirements.txt CHANGED
@@ -15,6 +15,7 @@ pydantic==2.11.5
15
  huggingface_hub==0.36.0
16
  datasets==3.0.1
17
  transformers==4.44.2
 
18
 
19
  # ONNX Runtime (๋ชจ๋ธ ์ถ”๋ก )
20
  onnxruntime==1.19.2
 
15
  huggingface_hub==0.36.0
16
  datasets==3.0.1
17
  transformers==4.44.2
18
+ tokenizers==0.19.1
19
 
20
  # ONNX Runtime (๋ชจ๋ธ ์ถ”๋ก )
21
  onnxruntime==1.19.2