Open sc-carson opened 10 months ago
使用ptuning进行微调后,在checkpoint-3000路径下并没有pytorch_model.bin,只有training_args.bin,无法运行prefix_state_dict = torch.load(os.path.join(CHECKPOINT_PATH, "pytorch_model.bin"))
No response
bash train.sh
- OS:Ubuntu 22.04 - Python:Python 3.10.13 - Transformers: - PyTorch:4.36.2 - CUDA Support (`python -c "import torch; print(torch.cuda.is_available())"`) :True
我也是,请问解决了吗
同,请问有无解决方案?
已解决 -> transforms库降到 requirements.txt 中的版本即可解决
还是不行
Is there an existing issue for this?
Current Behavior
使用ptuning进行微调后,在checkpoint-3000路径下并没有pytorch_model.bin,只有training_args.bin,无法运行prefix_state_dict = torch.load(os.path.join(CHECKPOINT_PATH, "pytorch_model.bin"))
Expected Behavior
No response
Steps To Reproduce
bash train.sh
Environment
Anything else?
No response