maxoul commited on
Commit
51ce61b
·
verified ·
1 Parent(s): 77f3bd7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -2
README.md CHANGED
@@ -2,12 +2,15 @@
2
  license: cc-by-nc-sa-4.0
3
  ---
4
 
 
 
 
5
  ```python
6
  from transformers import AutoModelForCausalLM, AutoModel
7
  import os
8
  import torch
9
- token = os.environ["token_id"]
10
- splade = AutoModelForCausalLM.from_pretrained("naver/splade-code-06B", token=token, trust_remote_code=True)
11
  device = (torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu"))
12
  splade.to(device)
13
  splade.eval()
 
2
  license: cc-by-nc-sa-4.0
3
  ---
4
 
5
+ SPLADE-Code-06B is a sparse retrieval model designed for code retrieval tasks. It is the top-performing models on MTEB for models below 1B (at time of writing, Feb 2026).
6
+
7
+
8
  ```python
9
  from transformers import AutoModelForCausalLM, AutoModel
10
  import os
11
  import torch
12
+
13
+ splade = AutoModelForCausalLM.from_pretrained("naver/splade-code-06B", trust_remote_code=True)
14
  device = (torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu"))
15
  splade.to(device)
16
  splade.eval()