VirusLM
Collection
pretraining on virus sqeuences such as SARS-CoV-2
•
1 item
•
Updated
from transformers import AutoTokenizer, AutoModelForCausalLM
checkpoint = "jiaxie/Hyena-SARS-CoV2"
tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)
print("Tokenizer Processed")
model_config = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True)
print("Model Processed")