m97j commited on
Commit
801bba3
ยท
1 Parent(s): f8d03f6

Initial codes commit

Browse files
Files changed (1) hide show
  1. models/initializer.py +8 -8
models/initializer.py CHANGED
@@ -5,13 +5,15 @@ from huggingface_hub import hf_hub_download
5
  from fastapi import FastAPI
6
  from config import HF_MODEL_REPO_ID, EMBED_MODEL, EMBED_DIR, RERANK_MODEL, RERANK_DIR
7
 
 
 
 
 
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
- trust_remote_code=True,
14
- force_download=True
15
  )
16
  embedder_model_path = hf_hub_download(
17
  repo_id=HF_MODEL_REPO_ID,
@@ -22,10 +24,8 @@ def initialize_models(app: FastAPI):
22
 
23
  # Reranker
24
  reranker_tokenizer = AutoTokenizer.from_pretrained(
25
- HF_MODEL_REPO_ID,
26
- subfolder=RERANK_DIR, # ํ† ํฌ๋‚˜์ด์ € ๊ด€๋ จ ํŒŒ์ผ์ด reranker/ ์•ˆ์— ์žˆ์œผ๋ฏ€๋กœ ์ง€์ •
27
- trust_remote_code=True,
28
- force_download=True
29
  )
30
  reranker_model_path = hf_hub_download(
31
  repo_id=HF_MODEL_REPO_ID,
 
5
  from fastapi import FastAPI
6
  from config import HF_MODEL_REPO_ID, EMBED_MODEL, EMBED_DIR, RERANK_MODEL, RERANK_DIR
7
 
8
+ # ๊ณต์‹ ๋ชจ๋ธ ๊ฒฝ๋กœ (์›๋ณธ ํ† ํฌ๋‚˜์ด์ €๋ฅผ ๋ถˆ๋Ÿฌ์˜ค๊ธฐ ์œ„ํ•ด ์ง€์ •)
9
+ EMBEDDER_ORIGINAL_ID = "Qwen/Qwen3-Embedding-0.6B"
10
+ RERANKER_ORIGINAL_ID = "Qwen/Qwen3-Reranker-0.6B"
11
+
12
  def initialize_models(app: FastAPI):
13
  # Embedder
14
  embedder_tokenizer = AutoTokenizer.from_pretrained(
15
+ EMBEDDER_ORIGINAL_ID,
16
+ trust_remote_code=True
 
 
17
  )
18
  embedder_model_path = hf_hub_download(
19
  repo_id=HF_MODEL_REPO_ID,
 
24
 
25
  # Reranker
26
  reranker_tokenizer = AutoTokenizer.from_pretrained(
27
+ RERANKER_ORIGINAL_ID,
28
+ trust_remote_code=True
 
 
29
  )
30
  reranker_model_path = hf_hub_download(
31
  repo_id=HF_MODEL_REPO_ID,