uer
/

pegasus-base-chinese-cluecorpussmall

Text2Text Generation

Inference Endpoints

Model card Files Files and versions Community

uer commited on Oct 17, 2023

Commit

0787554

•

1 Parent(s): 531062b

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -63,8 +63,8 @@ python3 pretrain.py --dataset_path cluecorpussmall_pegasus_seq512_dataset.pt \
 Finally, we convert the pre-trained model into Huggingface's format:
 ```
-python3 scripts/convert_pegasus_from_uer_to_huggingface.py --input_model_path cluecorpussmall_pegasus_base_seq512_model.bin-250000 \
-                                                           --output_model_path pytorch_model.bin \
                                                            --layers_num 12
 ```

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