You can deploy this model and start earning money today!

genbio-ai/
AIDO.Protein-16B-tcr_pmhc_affinity

Name: AIDO.Protein-16B-tcr_pmhc_affinity
Availability: InStock

Bo1015/tcr_pmhc_affinity

genbio-ai/AIDO.Protein-16B

finetune

other

Views

No views yet

LoRA fine-tuned checkpoint for tcr_pmhc_affinity prediction.

How to Use

Download model

python

from huggingface_hub import snapshot_download
from pathlib import Path

model_name = "genbio-ai/GB.Protein-16B-tcr_pmhc_affinity"
genbio_models_path = Path.home().joinpath('genbio_models', model_name)
genbio_models_path.mkdir(parents=True, exist_ok=True)
snapshot_download(repo_id=model_name, local_dir=genbio_models_path)

Load model for inference

python

import torch
from modelgenerator.tasks import SequenceClassification

ckpt_path = genbio_models_path.joinpath('model.ckpt')
model = SequenceClassification.load_from_checkpoint(ckpt_path, strict_loading=False).eval()

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

genbio-ai/AIDO.Protein-16B-tcr_pmhc_affinity

How to Use

Download model

Load model for inference

genbio-ai/
AIDO.Protein-16B-tcr_pmhc_affinity