Update README.md

README.md

@@ -52,8 +52,8 @@ def get_detailed_instruct(task_description: str, query: str) -> str:
     return f'{task_description}\nQuery: {query}'
 
 
-model = AutoModel.from_pretrained('yuri-no/gemma-argos', torch_dtype=torch.bfloat16).to('cuda')
-tokenizer = AutoTokenizer.from_pretrained('yuri-no/gemma-argos')
+model = AutoModel.from_pretrained('yuri-no/gemma-palm', torch_dtype=torch.bfloat16).to('cuda')
+tokenizer = AutoTokenizer.from_pretrained('yuri-no/gemma-palm')
 
 # Each query must come with a one-sentence instruction that describes the task
 task = 'Given a search query, retrieve relevant passages that answer the query'
@@ -80,7 +80,7 @@ embeddings = last_token_pool(outputs.last_hidden_state, batch_dict['attention_ma
 embeddings = F.normalize(embeddings, p=2, dim=1)
 scores = (embeddings[:2] @ embeddings[2:].T) * 100
 print(scores.tolist())
-# [[69.5, 30.5, 26.0],
-# [32.25, 66.5, 30.25]]
+# [[60.5, 16.375, 17.5],
+# [26.0, 58.25, 15.0625]]
 ```
 ---