Update README.md
Browse files
README.md
CHANGED
@@ -10,7 +10,7 @@ ProtST for binary localization.
|
|
10 |
The following script shows how to finetune ProtST on Gaudi.
|
11 |
|
12 |
## Running script
|
13 |
-
```
|
14 |
from transformers import AutoModel, AutoTokenizer, HfArgumentParser, TrainingArguments, Trainer
|
15 |
from transformers.data.data_collator import DataCollatorWithPadding
|
16 |
from transformers.trainer_pt_utils import get_parameter_names
|
|
|
10 |
The following script shows how to finetune ProtST on Gaudi.
|
11 |
|
12 |
## Running script
|
13 |
+
```diff
|
14 |
from transformers import AutoModel, AutoTokenizer, HfArgumentParser, TrainingArguments, Trainer
|
15 |
from transformers.data.data_collator import DataCollatorWithPadding
|
16 |
from transformers.trainer_pt_utils import get_parameter_names
|