Skip to content

Torch finally realised pickle is a bad idea, and told the entire world every time it loaded a model

License

Notifications You must be signed in to change notification settings

bitplane/torch_weightsonly

Repository files navigation

torch_weightsonly

So torch is spitting out errors to the command line in every app that doesn't use torch.load(weights_only=True)

Load this package instead of torch to fix that, and do it before you load anything else. Example:

import torch_weightsonly as torch
import whisper

# whisper won't give security grumbles

About

Torch finally realised pickle is a bad idea, and told the entire world every time it loaded a model

Resources

License

Stars

Watchers

Forks

Packages

No packages published