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

希望能恢复lora权重文件能生成.bin格式,新版本是.safetensors,希望能加入参数控制,llama.cpp的qlora要bin格式 #1957

Closed
1 task done
lizhiling12345 opened this issue Dec 22, 2023 · 1 comment
Labels
solved This problem has been already solved

Comments

@lizhiling12345
Copy link

Reminder

  • I have read the README and searched the existing issues.

Reproduction

python src/train_bash.py
--stage pt
--model_name_or_path /hy-tmp/model/AquilaCode-multi
--do_train
--dataset my_code
--template aquila
--lora_target all
--finetuning_type lora
--output_dir sft_checkpoint_aquila
--overwrite_cache
--per_device_train_batch_size 4
--gradient_accumulation_steps 1
--lr_scheduler_type cosine
--logging_steps 1
--save_steps 100
--save_total_limit 10
--learning_rate 5e-5
--num_train_epochs 1000
--plot_loss
--fp16
--quantization_bit 4

Expected behavior

No response

System Info

No response

Others

No response

@hiyouga
Copy link
Owner

hiyouga commented Dec 22, 2023

--save_safetensors False

@hiyouga hiyouga added the solved This problem has been already solved label Dec 22, 2023
@hiyouga hiyouga closed this as completed Dec 22, 2023
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

2 participants