TencentPretrain icon indicating copy to clipboard operation
TencentPretrain copied to clipboard

convert a tencentpretrain llama model to huggingface?

Open jamestch opened this issue 1 year ago • 3 comments

How to convert a tencentpretrain llama model to huggingface? can u share the script? Thanks a lot!

jamestch avatar Apr 06 '23 12:04 jamestch

你有找到这个方法吗

baketbek avatar Apr 19 '23 06:04 baketbek

去scripts/ 文件夹下找convert_tencentpretrain_to_llama.py,用这个转换。 python convert_tencentpretrain_to_llama.py --input_model_path 输入模型路径/input_model.bin --output_model_path 输出模型路径/pytorch_model.bin --layers_num 32 7B的模型层数是32

csho avatar May 10 '23 04:05 csho

https://github.com/huggingface/transformers/blob/main/src/transformers/models/llama/convert_llama_weights_to_hf.py

bmfire1 avatar May 18 '23 09:05 bmfire1