Skip to content

fix weights_only problem with pytorch#658

Open
Dronakurl wants to merge 1 commit intomarcoslucianops:masterfrom
Dronakurl:fixweightsonly
Open

fix weights_only problem with pytorch#658
Dronakurl wants to merge 1 commit intomarcoslucianops:masterfrom
Dronakurl:fixweightsonly

Conversation

@Dronakurl
Copy link

The export utilities only work with torch.load(..., weights_only=False)
See here

@feli-santos
Copy link

this fix works flawlessly

@alex-phoenixai
Copy link

+1

1 similar comment
@valentin-phoenix
Copy link
Contributor

+1

@valentin-phoenix
Copy link
Contributor

Work without this fix since torch>=2.8.0

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

Successfully merging this pull request may close these issues.

4 participants