Update src/embeddings.py
Browse files- src/embeddings.py +1 -1
src/embeddings.py
CHANGED
|
@@ -23,7 +23,7 @@ print(f"β
Using Hugging Face cache at {CACHE_DIR}")
|
|
| 23 |
# ----------------------------
|
| 24 |
try:
|
| 25 |
_model = SentenceTransformer(
|
| 26 |
-
"intfloat/e5-
|
| 27 |
cache_folder=CACHE_DIR
|
| 28 |
)
|
| 29 |
print("β
Loaded model: intfloat/e5-small-v2")
|
|
|
|
| 23 |
# ----------------------------
|
| 24 |
try:
|
| 25 |
_model = SentenceTransformer(
|
| 26 |
+
"intfloat/e5-large-v2", # β
Trained for retrieval-augmented QA
|
| 27 |
cache_folder=CACHE_DIR
|
| 28 |
)
|
| 29 |
print("β
Loaded model: intfloat/e5-small-v2")
|