SiddhJagani commited on
Commit
50cad82
·
verified ·
1 Parent(s): 0651e00

Update src/embedder.py

Browse files
Files changed (1) hide show
  1. 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():