Skip to content

add suport for baichuan2_7b - #131

Open
Yang-Yi20 wants to merge 10 commits into
Tencent:mainfrom
Yang-Yi20:yhy
Open

Yang-Yi20 wants to merge 10 commits into
Tencent:mainfrom
Yang-Yi20:yhy

Conversation

@Yang-Yi20

Copy link
Copy Markdown

convert

python3 scripts/convert_baichuan_from_huggingface_to_tencentpretrain.py --input_model_path $Baichuan2_7B_FOLDER --output_model_path models/baichuan2_7b.bin

test

python3 scripts/generate_lm.py --load_model_path models/baichuan2_7b.bin --spm_model_path $Baichuan2_7B_FOLDER/tokenizer.model --test_path beginning.txt --prediction_path generated_sentence.txt --config_path models/baichuan/7b_config.json

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

2 participants