correct links
Browse files
README.md
CHANGED
@@ -10,7 +10,7 @@ Note that the vocabulary size of this model (50275) differs from the original `l
|
|
10 |
|
11 |
`<|par|>, </|title|>, </|sec|>, <|sec-title|>, <|sent|>, <|title|>, <|abs|>, <|sec|>, </|sec-title|>, </|abs|>`.
|
12 |
|
13 |
-
Transferring the checkpoint weights and saving the model was done based on [this code](https://github.com/dwadden/multivers/blob/
|
14 |
```python
|
15 |
import os
|
16 |
import pathlib
|
@@ -56,7 +56,7 @@ if not os.path.exists(model_dir):
|
|
56 |
model.save_pretrained(model_dir)
|
57 |
```
|
58 |
|
59 |
-
The tokenizer was resized and saved following [this code](https://github.com/dwadden/multivers/blob/
|
60 |
```python
|
61 |
from transformers import AutoTokenizer
|
62 |
|
|
|
10 |
|
11 |
`<|par|>, </|title|>, </|sec|>, <|sec-title|>, <|sent|>, <|title|>, <|abs|>, <|sec|>, </|sec-title|>, </|abs|>`.
|
12 |
|
13 |
+
Transferring the checkpoint weights and saving the model was done based on [this code](https://github.com/dwadden/multivers/blob/main/multivers/model.py#L145) from the MultiVerS repository, the versions of `transformers==4.2.2` and `torch==1.7.1` correspond to the MultiVerS [requirements.txt](https://github.com/dwadden/multivers/blob/main/requirements.txt):
|
14 |
```python
|
15 |
import os
|
16 |
import pathlib
|
|
|
56 |
model.save_pretrained(model_dir)
|
57 |
```
|
58 |
|
59 |
+
The tokenizer was resized and saved following [this code](https://github.com/dwadden/multivers/blob/main/multivers/data.py#L14) from the MultiVerS repository:
|
60 |
```python
|
61 |
from transformers import AutoTokenizer
|
62 |
|