uer
/

gpt2-chinese-poem

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

uer commited on Oct 17, 2023

Commit

6335c88

•

1 Parent(s): 2aaa081

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -71,7 +71,7 @@ python3 pretrain.py --dataset_path poem_dataset.pt \
 Finally, we convert the pre-trained model into Huggingface's format:
 ```
-python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path models/poem_gpt2_model.bin -200000 \
                                                         --output_model_path pytorch_model.bin \
                                                         --layers_num 12
 ```

 Finally, we convert the pre-trained model into Huggingface's format:
 ```
+python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path models/poem_gpt2_model.bin-200000 \
                                                         --output_model_path pytorch_model.bin \
                                                         --layers_num 12
 ```