Views
No views yet
pip install lucagplm1from lucagplm import LucaGPLMModel, LucaGPLMTokenizer
2
3# Load model
4model = LucaGPLMModel.from_pretrained("Yuanfei/LucaVirus-default-step3.8M")
5tokenizer = LucaGPLMTokenizer.from_pretrained("Yuanfei/LucaVirus-default-step3.8M")
6
7# Example usage
8seq = "ATCG"
9inputs = tokenizer(seq, seq_type="gene",return_tensors="pt")
10outputs = model(**inputs)
11
12seq = "NSQTA"
13inputs = tokenizer(seq, seq_type="prot",return_tensors="pt")
14outputs = model(**inputs)
15
16print(outputs.last_hidden_state.shape)