[BUG] <title> Unable to load trained LoRa model weights using AutoPeftModelForCausalLM.from_pretrained()
jweihe opened this issue · comments
是否已有关于该错误的issue或讨论? | Is there an existing issue / discussion for this?
- 我已经搜索过已有的issues和讨论 | I have searched the existing issues / discussions
该问题是否在FAQ中有解答? | Is there an existing answer for this in FAQ?
- 我已经搜索过FAQ | I have searched FAQ
当前行为 | Current Behavior
I encountered an issue while trying to load the weights of a trained LoRa model using the AutoPeftModelForCausalLM.from_pretrained() method. The error message indicates that the tokenization_qwen.py file could not be located in the specified checkpoint directory.
Here is the error message I received:
Could not locate the tokenization_qwen.py inside output_model_qwen_hr-llm-vit/checkpoint OSError:output_model_qwen_hr-llm-vit/checkpoint does not appear to have a file named tokenization_qwen.py.
The code I used to load the model is:
model = AutoPeftModelForCausalLM.from_pretrained(args.checkpoint, device_map='cuda', trust_remote_code=True).eval()
I would appreciate any guidance on how to resolve this issue and successfully load the trained LoRa model weights.
期望行为 | Expected Behavior
No response
复现方法 | Steps To Reproduce
No response
运行环境 | Environment
- OS:
- Python:
- Transformers:
- PyTorch:
- CUDA (`python -c 'import torch; print(torch.version.cuda)'`):
备注 | Anything else?
No response