- Notifications
You must be signed in to change notification settings - Fork328
bind with custom python ML model#509
-
Hello, I love what you guys have done. I am wondering whether it's possible to bind my own custom model built in PyTorch. I am looking to make a recommendation system. If so, is there any guide or API template that I should follow? Thanks! |
BetaWas this translation helpful?Give feedback.
All reactions
If you’re able to publish your PyTorch model via huggingface, you can load it in v1:https://postgresml.org/user_guides/transformers/pre_trained_models/
We’re still working to port the huggingface APIs to rust in v2 (PRs welcome :)
The issue is that models require some metadata etc. We’re looking at some different ways to incorporate deep learning APIs ergonomically in v2, which are more complex than other more traditional algorithms. We’ll get there sooner or later.
Replies: 1 comment
-
If you’re able to publish your PyTorch model via huggingface, you can load it in v1:https://postgresml.org/user_guides/transformers/pre_trained_models/ We’re still working to port the huggingface APIs to rust in v2 (PRs welcome :) The issue is that models require some metadata etc. We’re looking at some different ways to incorporate deep learning APIs ergonomically in v2, which are more complex than other more traditional algorithms. We’ll get there sooner or later. |
BetaWas this translation helpful?Give feedback.