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.25k stars 104 forks source link

Add support for loading models via safetensors #186

Closed vrdn-23 closed 4 hours ago

vrdn-23 commented 1 week 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 1 week ago

looking forward to

vrdn-23 commented 3 days ago

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

urchade commented 3 days ago

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