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

Share Tensor Error for saving model #3

Open
Joseph-dth opened this issue Nov 14, 2024 · 1 comment
Open

Share Tensor Error for saving model #3

Joseph-dth opened this issue Nov 14, 2024 · 1 comment

Comments

@Joseph-dth
Copy link

Truly appreciate for sharing this impressive work!

I've been trying out the merging function and encountered some tensor sharing errors with the latest transformers package. I found that using transformers 4.35.x allows the code to run (with some warnings) while newer versions produce sharing tensor errors.

Would you mind to specify the exact dependencies' version of your environment?

@krP471
Copy link

krP471 commented Dec 13, 2024

You can try with safe_serialization=False in save_pretrained function.

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

2 participants