mlem icon indicating copy to clipboard operation
mlem copied to clipboard

Support huggingface transformers models and datasets

Open aguschin opened this issue 3 years ago • 7 comments

We need to add support for the huggingface transformers models and datasets.

aguschin avatar May 23 '22 09:05 aguschin

Are there any predictions for timelines on when this might be implemented?

judahrand avatar May 22 '23 15:05 judahrand

@judahrand could you please describe the use case a bit more? so that's easier for us to prioritize this. thanks!

shcheklein avatar May 22 '23 18:05 shcheklein

@judahrand could you please describe the use case a bit more? so that's easier for us to prioritize this. thanks!

We use huggingface transformer models and would prefer to save them in their native form so that they can be picked up by BentoML for deployment.

judahrand avatar May 22 '23 20:05 judahrand

I looks like work was started on this: #553

judahrand avatar May 23 '23 07:05 judahrand

@judahrand, unfortunately, I'm at capacity now. I'd love to see this implemented, but don't have time 😞

Why you need this implemented specifically in MLEM? Looks like simply saving it with HF + using Bento would work.

aguschin avatar May 24 '23 08:05 aguschin

@judahrand, unfortunately, I'm at capacity now. I'd love to see this implemented, but don't have time 😞

Why you need this implemented specifically in MLEM? Looks like simply saving it with HF + using Bento would work.

We would like to use MLEM to power our 'model registry' and then deploy those models using BentoML. I suppose we can save the model to the model registry using the standard HF implementation and manually track it using DVC but we'd like to use MLEM if possible.

judahrand avatar May 25 '23 08:05 judahrand

Thanks @judahrand! To better understand your use case: so you are looking not for model deployment only, but for "model registry" kind of things as well (like managing multiple models, registering versions, etc). Did you take a look at our docs for model registry already? https://dvc.org/doc/use-cases/model-registry

aguschin avatar May 25 '23 12:05 aguschin