Skip to content

Conversation

@antonallote
Copy link

Replace pickle serialization with safetensors to address security concerns with arbitrary code execution during model loadin

  • add SafetensorsSerializer that stores tensors via safetensors and metadata as JSON
  • Model.save() now uses safetensors per default (use_safetensors=True)
  • Auto-detect format on load with deprecation warning for pickle models

Closes #3670

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.

[Feature]: Change model serialization

1 participant