jiaxie
/

DeepProtT5-Fluorescence

Model card Files Files and versions Community

jiaxie commited on Mar 8

Commit

e938d9b

·

verified ·

1 Parent(s): aaae04d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -58,5 +58,5 @@ class T5RegressionModel(PreTrainedModel):
 Then we load our pretrained model
 ```python
 tokenizer = T5Tokenizer.from_pretrained("jiaxie/DeepProtT5-Fluorescence", do_lower_case=False)
-model = T5ClassificationModel.from_pretrained("jiaxie/DeepProtT5-Fluorescence ", torch_dtype=torch.bfloat16).to("cuda")
 ```

 Then we load our pretrained model
 ```python
 tokenizer = T5Tokenizer.from_pretrained("jiaxie/DeepProtT5-Fluorescence", do_lower_case=False)
+model = T5RegressionModel.from_pretrained("jiaxie/DeepProtT5-Fluorescence", torch_dtype=torch.bfloat16).to("cuda")
 ```