search

alvarobartt / safejax

Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
https://alvarobartt.github.io/safejax/
MIT License
42 stars 5 forks source link

Add `metadata` as a field in `serialize` #22

Closed alvarobartt closed 1 year ago

alvarobartt commented 1 year ago

safetensors.flax.save and safetensors.flax.save_file provide another param besides tensors which is metadata, a dictionary with some metadata to be stored too.

More information at https://huggingface.co/docs/safetensors/api/flax#safetensors.flax.save and https://huggingface.co/docs/safetensors/api/flax#safetensors.flax.save_file

alvarobartt commented 1 year ago

More information at https://github.com/huggingface/safetensors/issues/147