Add support for loading models via safetensors

urchade / GLiNER

Generalist and Lightweight Model for Named Entity Recognition (Extract any entity types from texts) @ NAACL 2024

https://arxiv.org/abs/2311.08526

Apache License 2.0

1.46k stars 126 forks source link

Add support for loading models via safetensors #186

Closed vrdn-23 closed 2 months ago

vrdn-23 commented 2 months ago

Thanks a lot for the awesome project! Loading models through .bin executables are inherently risky and can lead to malicious code execution. Please consider adding support for loading the safetensors variant of the models provided.

If this is something that you would be interested in, I'd be happy to help out with a PR!

xingchaozh commented 2 months ago

looking forward to

vrdn-23 commented 2 months ago

@urchade I've raised a PR to fix the issue. Let me know if you have any comments!

urchade commented 2 months ago

Thank you @vrdn-23, will review it soon!

vrdn-23 commented 2 months ago

@urchade Thanks for reviewing! Is there an ETA on when the next release is going to be cut?

urchade commented 2 months ago

Hey @vrdn-23, will do it tomorrow!