Downloads · 30 days
16
6% of all-time downloads
dnafm/144M_H_MLM_last
144M_H_MLM_last is a machine learning model from dnafm. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Downloads · 30 days
16
6% of all-time downloads
All-time downloads
282
Public
Parameters
145M
579 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors579 MB · 100%
From the Hugging Face model README
Load the model:
import torch
from transformers import AutoModel, AutoTokenizer
model_name = "rnalm/144M_H_MLM_last"
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
# Move model to GPU
model = model.cuda()
Get embeddings:
inputs = tokenizer("ACGTACGT", return_tensors="pt")
with torch.no_grad():
outputs = model(input_ids=inputs["input_ids"].cuda())
outputs.last_hidden_state.shape
# torch.Size([1, 8, 768])
outputs.seq_logits.shape
# torch.Size([1, 8, 11])