Spaces:
Sleeping
Sleeping
Update src/embedder.py
Browse files- src/embedder.py +2 -2
src/embedder.py
CHANGED
|
@@ -12,8 +12,8 @@ print("Loading ModernBERT Embed (CPU version)...")
|
|
| 12 |
# e.g. nomic-ai/nomic-embed-text-v1.5 or sentence-transformers/all-MiniLM-L6-v2
|
| 13 |
MODEL_NAME = "nomic-ai/nomic-embed-text-v1.5"
|
| 14 |
|
| 15 |
-
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
| 16 |
-
model = AutoModel.from_pretrained(MODEL_NAME)
|
| 17 |
model.eval()
|
| 18 |
|
| 19 |
if torch.cuda.is_available():
|
|
|
|
| 12 |
# e.g. nomic-ai/nomic-embed-text-v1.5 or sentence-transformers/all-MiniLM-L6-v2
|
| 13 |
MODEL_NAME = "nomic-ai/nomic-embed-text-v1.5"
|
| 14 |
|
| 15 |
+
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
|
| 16 |
+
model = AutoModel.from_pretrained(MODEL_NAME, trust_remote_code=True)
|
| 17 |
model.eval()
|
| 18 |
|
| 19 |
if torch.cuda.is_available():
|