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

[BUG] <title> Unable to load trained LoRa model weights using AutoPeftModelForCausalLM.from_pretrained() #379

Open
2 tasks done
jweihe opened this issue May 9, 2024 · 0 comments

Comments

@jweihe
Copy link

jweihe commented May 9, 2024

是否已有关于该错误的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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant