SamuelYang commited on
Commit
2e0194a
·
verified ·
1 Parent(s): b3a1df9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -6,7 +6,6 @@ language:
6
  - zh
7
  ---
8
  # INF-Retriever-v1
9
- ## Model Overview
10
  - **INF-Retriever-v1** is an LLM-based dense retrieval model developed by [INF TECH](https://www.infly.cn/en).
11
  It is built upon the [gte-Qwen2-7B-instruct](https://huggingface.co/Alibaba-NLP/gte-Qwen2-7B-instruct) model and specifically fine-tuned to excel in retrieval tasks, particularly for Chinese and English data.
12
 
@@ -71,6 +70,7 @@ embeddings = last_token_pool(outputs.last_hidden_state, batch_dict['attention_ma
71
  embeddings = F.normalize(embeddings, p=2, dim=1)
72
  scores = (embeddings[:2] @ embeddings[2:].T) * 100
73
  print(scores.tolist())
 
74
  ```
75
 
76
  ## Evaluation
 
6
  - zh
7
  ---
8
  # INF-Retriever-v1
 
9
  - **INF-Retriever-v1** is an LLM-based dense retrieval model developed by [INF TECH](https://www.infly.cn/en).
10
  It is built upon the [gte-Qwen2-7B-instruct](https://huggingface.co/Alibaba-NLP/gte-Qwen2-7B-instruct) model and specifically fine-tuned to excel in retrieval tasks, particularly for Chinese and English data.
11
 
 
70
  embeddings = F.normalize(embeddings, p=2, dim=1)
71
  scores = (embeddings[:2] @ embeddings[2:].T) * 100
72
  print(scores.tolist())
73
+ # [[86.87025451660156, 67.82366180419922], [59.510135650634766, 82.33667755126953]]
74
  ```
75
 
76
  ## Evaluation