How to register Transformer Model in MLFLOW Model Registery

Hi All,

I am new to Transformer. I am trying to build pipeline with MLFlow. I am able to track the progress in MLFlow but MLFlow is saving the transformer file as artifact not as a model because of which I am not able to register it in MLFLOW.

My Idea is like we do with sklearn model in MLFLOW we annotate it as Staging or Production. Same I want to do with Transformer Model since Transformers Files are saved as Artifacts not able to create Model Registry in MLFOW. Is there any way to do it…?

Regards
Nabarun Barua

1 Like