huggingface / exporters

Export Hugging Face models to Core ML and TensorFlow Lite

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Support for .safetensors files?

hg1722 opened this issue · comments

When trying to convert a model with safetensor weights, exporters fails with [MODEL] does not appear to have a file named pytorch_model.bin, tf_model.h5, model.ckpt or flax_model.msgpack. Adding support would help out a lot, especially as safetensors seem to be pushed as the new standard for storing weights.

Hi @hg1722, 100% agree!

Did you check that it fails when you have safetensors installed? What model are you using to test, if you can share?