search

SeldonIO / MLServer

An inference server for your machine learning models, including support for multiple frameworks, multi-model serving and more
https://mlserver.readthedocs.io/en/latest/
Apache License 2.0
685 stars 178 forks source link

Lightning Flash examples #291

Open tchaton opened 3 years ago

tchaton commented 3 years ago

Hey there,

Awesome framework. Would be nice to have a PyTorch Lightning or a Flash example (https://github.com/PyTorchLightning/lightning-flash).

Best, T.C

adriangonz commented 2 years ago

Hey @tchaton,

That would be really cool!

Would the loading of Lightning / Flash models be different to regular PyTorch models? Is there any particular feature of Lightning that you think would make sense to leverage in MLServer?

I think we're both based in the UK, so would be great to jump on a call to share some ideas and learn more about Lightning. I'll send you a separate email with some available times (I can just use the email listed in your GH profile - if that's alright).