pantoniadis commited on
Commit
b4df2d6
·
verified ·
1 Parent(s): e2f9575

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -3
README.md CHANGED
@@ -19,10 +19,8 @@ model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
19
  model = model.cuda()
20
  ```
21
 
22
- Inference without using the track prediction head:
23
  ```
24
- # disable the track head in order to avoid providing the metadata
25
- model.model.predict_tracks = False
26
  inputs = tokenizer("ACGTACGT", return_tensors="pt")
27
 
28
  with torch.no_grad():
 
19
  model = model.cuda()
20
  ```
21
 
22
+ Get embeddings:
23
  ```
 
 
24
  inputs = tokenizer("ACGTACGT", return_tensors="pt")
25
 
26
  with torch.no_grad():