Habana
/

bert-large-uncased-whole-word-masking

Model card Files Files and versions Community

regisss HF staff commited on Jun 4, 2022

Commit

ba8e8b6

•

1 Parent(s): 2ffc790

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ The model is instantiated the same way as in the Transformers library.
 The only difference is that there are a few new training arguments specific to HPUs:
 ```
-from optimum.habana import GaudiConfig, GaudiTrainer, GaudiTrainingArguments
 from transformers import BertTokenizer, BertModel
 tokenizer = BertTokenizer.from_pretrained("bert-large-uncased-whole-word-masking")

 The only difference is that there are a few new training arguments specific to HPUs:
 ```
+from optimum.habana import GaudiTrainer, GaudiTrainingArguments
 from transformers import BertTokenizer, BertModel
 tokenizer = BertTokenizer.from_pretrained("bert-large-uncased-whole-word-masking")