microsoft / dstoolkit-mlops-v2

This repository contains the basic repository structure for machine learning projects based on Azure technologies (Azure ML and Azure DevOps).
Other
13 stars 3 forks source link

Add support for the PyTorch Custom model into dstoolkit #127

Open sweanan opened 2 months ago

sweanan commented 2 months ago

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:

vilcek commented 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.