Hugging Face's logo

charlieoneill
/
my_modernbert_person_embeddings

Feature Extraction
PyTorch
Model card Files Community
charlieoneill commited on
Commit
b3ae512
·
verified ·
1 Parent(s): 3d541d7

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -26,7 +26,7 @@ class PersonEmbeddings(nn.Module):
26
  class CustomEmbeddingPipeline:
27
  def __init__(self, model_id="answerdotai/ModernBERT-base"):
28
  # Load your base tokenizer
29
- self.tokenizer = AutoTokenizer.from_pretrained(model_id)
30
  # Load your PersonEmbeddings
31
  self.model = PersonEmbeddings(model_id)
32
  ckpt_path = "pytorch_model.bin"
 
26
  class CustomEmbeddingPipeline:
27
  def __init__(self, model_id="answerdotai/ModernBERT-base"):
28
  # Load your base tokenizer
29
+ self.tokenizer = AutoTokenizer.from_pretrained("charlieoneill/my_modernbert_person_embeddings")
30
  # Load your PersonEmbeddings
31
  self.model = PersonEmbeddings(model_id)
32
  ckpt_path = "pytorch_model.bin"