lonelycrab88 commited on
Commit
7400256
·
verified ·
1 Parent(s): fd4e32e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -28,8 +28,8 @@ from model.UBL_utils import CodonLogitsProcessor
28
 
29
  from transformers.generation.logits_process import LogitsProcessorList
30
 
31
- tokenizer = UBSLMTokenizer.from_pretrained("lonelycrab88/BiooBang-1.0")
32
- model = UniBioseqForCausalLM.from_pretrained("lonelycrab88/BiooBang-1.0", device_map='auto')
33
 
34
  protein_prompt = "MASSDKQTSPKPPPSPSPLRNSKFCQSNMRILIS"
35
  input_ids = torch.tensor([tokenizer.encode(input_protein)+[36]]).to(model.device)
 
28
 
29
  from transformers.generation.logits_process import LogitsProcessorList
30
 
31
+ tokenizer = UBSLMTokenizer.from_pretrained("lonelycrab88/BiooBang-1.0-HEK293T")
32
+ model = UniBioseqForCausalLM.from_pretrained("lonelycrab88/BiooBang-1.0-HEK293T", device_map='auto')
33
 
34
  protein_prompt = "MASSDKQTSPKPPPSPSPLRNSKFCQSNMRILIS"
35
  input_ids = torch.tensor([tokenizer.encode(input_protein)+[36]]).to(model.device)