Initial codes commit
Browse files- 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 |
-
|
| 12 |
-
|
| 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 |
-
|
| 26 |
-
|
| 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,
|