Open sweanan opened 2 months ago
The idea is to integrate an example of a custom Gen AI model. We are suggesting adding an example using a model from Hugging Face Transformers library, which provides access to hundreds of models for NLP, computer vision, audio, and others, through a common API. That will make this new example potentially reusable across many use cases.
Description:
DS Toolkit currently has samples for MLFlow model, in case of integrating the custom models, there are changes needed to integrate that into dstoolkit. Add support for this with a general Pytorch custom models
Acceptance criteria: