Michielo commited on
Commit
302a420
·
verified ·
1 Parent(s): 4a5e650

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -21,14 +21,14 @@ widget:
21
 
22
  The mt5-small nl-en translation model is a finetuned version of [google/mt5-small](https://huggingface.co/google/mt5-small).
23
 
24
- It was finetuned on 237k rows of the [iwslt2017](https://huggingface.co/datasets/iwslt2017/viewer/iwslt2017-en-nl) dataset and roughly 38k rows of the [opus_books](https://huggingface.co/datasets/opus_books/viewer/en-nl) dataset. The model was trained for 15 epochs with a batchsize of 8 and is currently **not** considered finished.
25
 
26
 
27
  ## How to use
28
 
29
  **Install dependencies**
30
  ```bash
31
- pip install transformers
32
  ```
33
 
34
  You can use the following code for model inference. This model was finetuned to work with an identifier when prompted that needs to be present for the best results.
 
21
 
22
  The mt5-small nl-en translation model is a finetuned version of [google/mt5-small](https://huggingface.co/google/mt5-small).
23
 
24
+ It was finetuned on 237k rows of the [iwslt2017](https://huggingface.co/datasets/iwslt2017/viewer/iwslt2017-en-nl) dataset and roughly 38k rows of the [opus_books](https://huggingface.co/datasets/opus_books/viewer/en-nl) dataset. The model was trained in multiple phases with different epochs & batch sizes.
25
 
26
 
27
  ## How to use
28
 
29
  **Install dependencies**
30
  ```bash
31
+ pip install transformers, sentencepiece, protobuf
32
  ```
33
 
34
  You can use the following code for model inference. This model was finetuned to work with an identifier when prompted that needs to be present for the best results.