diff --git a/export_model.py b/export_model.py
index 0522eae..553fd2d 100644
--- a/export_model.py
+++ b/export_model.py
@@ -96,6 +96,7 @@ def to_save(x):
             mlflow.set_tracking_uri(args.mlflow_url)
             mlflow.set_experiment('model_export')
 
+            mlflow.log_artifact(str(model_path / 'config.yaml'), artifact_path='model_onnx')
             mlflow.onnx.log_model(onnx_model, artifact_path='model_onnx', registered_model_name=args.mlflow_model_name)