Why MLflow and Databricks library versions?
Did you try increasing the MLflow artifact upload/download timeout, for example:
import os
os.environ["MLFLOW_ARTIFACT_UPLOAD_DOWNLOAD_TIMEOUT"] = "120"
Wiliam Rosa
Data Engineer | Machine Learning Engineer
LinkedIn: linkedin.com/in/wiliamrosa