shirayukikun
commited on
Update sample code
Browse files
README.md
CHANGED
@@ -16,7 +16,7 @@ language:
|
|
16 |
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
17 |
|
18 |
model = AutoModelForMaskedLM.from_pretrained(
|
19 |
-
"tohoku-nlp/tohokunlp-bert-500m-
|
20 |
)
|
21 |
tokenizer = AutoTokenizer.from_pretrained("tohoku-nlp/tohokunlp-bert-500m-sq4096-alpha")
|
22 |
```
|
|
|
16 |
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
17 |
|
18 |
model = AutoModelForMaskedLM.from_pretrained(
|
19 |
+
"tohoku-nlp/tohokunlp-bert-500m-sq8192-alpha", trust_remote_code=True
|
20 |
)
|
21 |
tokenizer = AutoTokenizer.from_pretrained("tohoku-nlp/tohokunlp-bert-500m-sq4096-alpha")
|
22 |
```
|