YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
Load model for inference
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))
- Downloads last month
- 0