Updating README with 2.0 in name
Browse files
README.md
CHANGED
|
@@ -40,8 +40,8 @@ This model currently needs a custom wrapper from `modeling_ltgbert.py`, you shou
|
|
| 40 |
import torch
|
| 41 |
from transformers import AutoTokenizer, AutoModelForMaskedLM
|
| 42 |
|
| 43 |
-
tokenizer = AutoTokenizer.from_pretrained("HPLT/
|
| 44 |
-
model = AutoModelForMaskedLM.from_pretrained("HPLT/
|
| 45 |
|
| 46 |
mask_id = tokenizer.convert_tokens_to_ids("[MASK]")
|
| 47 |
input_text = tokenizer("It's a beautiful[MASK].", return_tensors="pt")
|
|
@@ -60,13 +60,13 @@ We are releasing 10 intermediate checkpoints for each model at intervals of ever
|
|
| 60 |
|
| 61 |
You can load a specific model revision with `transformers` using the argument `revision`:
|
| 62 |
```python
|
| 63 |
-
model = AutoModelForMaskedLM.from_pretrained("HPLT/
|
| 64 |
```
|
| 65 |
|
| 66 |
You can access all the revisions for the models with the following code:
|
| 67 |
```python
|
| 68 |
from huggingface_hub import list_repo_refs
|
| 69 |
-
out = list_repo_refs("HPLT/
|
| 70 |
print([b.name for b in out.branches])
|
| 71 |
```
|
| 72 |
|
|
|
|
| 40 |
import torch
|
| 41 |
from transformers import AutoTokenizer, AutoModelForMaskedLM
|
| 42 |
|
| 43 |
+
tokenizer = AutoTokenizer.from_pretrained("HPLT/hplt_bert_base_2_0_eng-Latn")
|
| 44 |
+
model = AutoModelForMaskedLM.from_pretrained("HPLT/hplt_bert_base_2_0_eng-Latn", trust_remote_code=True)
|
| 45 |
|
| 46 |
mask_id = tokenizer.convert_tokens_to_ids("[MASK]")
|
| 47 |
input_text = tokenizer("It's a beautiful[MASK].", return_tensors="pt")
|
|
|
|
| 60 |
|
| 61 |
You can load a specific model revision with `transformers` using the argument `revision`:
|
| 62 |
```python
|
| 63 |
+
model = AutoModelForMaskedLM.from_pretrained("HPLT/hplt_bert_base_2_0_eng-Latn", revision="step21875", trust_remote_code=True)
|
| 64 |
```
|
| 65 |
|
| 66 |
You can access all the revisions for the models with the following code:
|
| 67 |
```python
|
| 68 |
from huggingface_hub import list_repo_refs
|
| 69 |
+
out = list_repo_refs("HPLT/hplt_bert_base_2_0_eng-Latn")
|
| 70 |
print([b.name for b in out.branches])
|
| 71 |
```
|
| 72 |
|