mesolitica
/

llama2-embedding-2b-8k-contrastive

Feature Extraction

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

huseinzol05 commited on Dec 12, 2023

Commit

26e5367

•

1 Parent(s): 6082445

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -19,8 +19,8 @@ WandB report, https://wandb.ai/mesolitica/llama2-embedding-600m-contrastive/repo
 from transformers import AutoModel, AutoTokenizer
 from sklearn.metrics.pairwise import cosine_similarity
-model = AutoModel.from_pretrained('llama2-embedding-600m-8k-contrastive', trust_remote_code = True)
-tokenizer = AutoTokenizer.from_pretrained('llama2-embedding-600m-8k-contrastive')
 input_ids = tokenizer(
     [

 from transformers import AutoModel, AutoTokenizer
 from sklearn.metrics.pairwise import cosine_similarity
+model = AutoModel.from_pretrained('llama2-embedding-2b-8k-contrastive', trust_remote_code = True)
+tokenizer = AutoTokenizer.from_pretrained('llama2-embedding-2b-8k-contrastive')
 input_ids = tokenizer(
     [