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…?

Nabarun Barua