File size: 372 Bytes
b3b2667
 
 
45b80de
 
fa556f2
 
 
 
 
 
45b80de
fa556f2
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
---
license: other
---
## How to Use
### Load model for inference
```python
import torch
from transformers import AutoModel

model = AutoModel.from_pretrained("genbio-ai/dummy-ckpt-hf", trust_remote_code=True)

collated_batch = model.genbio_model.collate({"sequences": ["ACGT", "AGCT"]})
logits = model(collated_batch)
print(logits)
print(torch.argmax(logits, dim=-1))
```