Update README.md
Browse files
README.md
CHANGED
@@ -74,7 +74,7 @@ Finally, we convert the pre-trained model into Huggingface's format:
|
|
74 |
```
|
75 |
python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path poem_gpt_base_model.bin \
|
76 |
--output_model_path pytorch_model.bin \
|
77 |
-
--
|
78 |
```
|
79 |
|
80 |
### BibTeX entry and citation info
|
|
|
74 |
```
|
75 |
python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path poem_gpt_base_model.bin \
|
76 |
--output_model_path pytorch_model.bin \
|
77 |
+
--layers_num 12
|
78 |
```
|
79 |
|
80 |
### BibTeX entry and citation info
|