Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ValueError: Target modules ['q_proj', 'v_proj'] not found in the base model. Please check the target modules and try again. #43

Closed
su-heyang opened this issue Jun 17, 2023 · 2 comments
Labels
solved This problem has been already solved

Comments

@su-heyang
Copy link

train_sft.py训练指令:
CUDA_VISIBLE_DEVICES=0 python src/train_sft.py
--model_name_or_path /data1/projects/baichuan-7B/
--do_train
--dataset alpaca_gpt4_zh
--finetuning_type lora
--output_dir output
--overwrite_cache
--per_device_train_batch_size 4
--gradient_accumulation_steps 4
--lr_scheduler_type cosine
--logging_steps 10
--save_steps 1000
--learning_rate 5e-5
--num_train_epochs 3.0
--plot_loss
--fp16

训练报错ValueError: Target modules ['q_proj', 'v_proj'] not found in the base model. Please check the target modules and try again.
value error

有没有大佬知道怎么解决,谢谢!

@intothephone
Copy link

加参数:--lora_target W_pack

@hiyouga hiyouga added the solved This problem has been already solved label Jun 19, 2023
@hiyouga hiyouga closed this as completed Jun 19, 2023
@1ring2rta
Copy link

百川的attention应该是把(Wq,Wk,Wv) concat成一个W_pack了

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
solved This problem has been already solved
Projects
None yet
Development

No branches or pull requests

4 participants