Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

utility to auto download and auto convert a model? #409

Open
jondot opened this issue Aug 13, 2023 · 1 comment
Open

utility to auto download and auto convert a model? #409

jondot opened this issue Aug 13, 2023 · 1 comment

Comments

@jondot
Copy link
Contributor

jondot commented Aug 13, 2023

Hi, i'm wondering if building a utility that auto downloads any model if missing, similar to the python lib, (and caches it) but also notices there's no ot file, and will run a conversion, if that kind of thing is feasible or there are too many case-by-case evaluation issues where this has to be a manual process?

@guillaume-be
Copy link
Owner

Hello @jondot ,

The library contains a helper script (Python based) to convert a Pytorch weights file to the C-array expected by this crate. Unfortunately Pytorch serialization uses Pickle by default meaning a Python installation is required to open these archive and convert them to a raw format.

The automated download utility exists to point to already-convert model files (via RemoteResource). Combining both would require starting a Python process within the Rust application which is not really desirable in my opinion.

What do you think?

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

No branches or pull requests

2 participants