| # 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", "ACGT"]}) | |
| logits = model(collated_batch) | |
| print(logits) | |
| print(torch.argmax(logits, dim=-1)) | |
| ``` |