Transformers How to use Moonn1205/AntigenLM-prediction-sequence with Transformers:
# Load model directly
from transformers import AutoTokenizer, GPTForFluMultiTask
tokenizer = AutoTokenizer.from_pretrained("Moonn1205/AntigenLM-prediction-sequence")
model = GPTForFluMultiTask.from_pretrained("Moonn1205/AntigenLM-prediction-sequence", device_map="auto") 
