Mlflow with Hugging Face

I am trying to load models to mlflow, but I keep getting: NO ARTIFACT RECORDED.

I am using the attribute “callbacks = [MLflowCallback]” of the Trainer function, but when training, only the metrics are saved but not the artifacts.

Someone has managed to get artifacts loaded also using transformers.

Thanks in advance :slight_smile: