SaeedLab
/

ProtEnrich-ProGen2

Feature Extraction

Model card Files Files and versions

gabrielbianchin commited on 7 days ago

Commit

4b265b0

·

1 Parent(s): eef3da7

upload

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -9,7 +9,7 @@ encoder = AutoModelForCausalLM.from_pretrained("hugohrban/progen2-base",  trust_
 protenrich = AutoModel.from_pretrained("SaeedLab/ProtEnrich-ProGen2", trust_remote_code=True)
 seqs = ["MKTFFVLLL"]
-inputs = tokenizer(seqs, return_tensors="pt", padding=True)
 with torch.no_grad():
   outputs = encoder(**inputs, output_hidden_states=True)

 protenrich = AutoModel.from_pretrained("SaeedLab/ProtEnrich-ProGen2", trust_remote_code=True)
 seqs = ["MKTFFVLLL"]
+inputs = tokenizer(seqs, return_tensors="pt")
 with torch.no_grad():
   outputs = encoder(**inputs, output_hidden_states=True)