Downloads · 30 days
5
4% of all-time downloads
DianLiI/dummy-ckpt-hf
dummy-ckpt-hf is a machine learning model from DianLiI. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Downloads · 30 days
5
4% of all-time downloads
All-time downloads
121
Public
Parameters
4.5M
18.2 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors18.2 MB · 100%
From the Hugging Face model README
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))