跳至主内容

MLflow 2.4.0

· 3 min read
MLflow maintainers
MLflow 维护者
MLflow maintainers

MLflow 2.4.0 包括多个主要功能和改进

功能：

  • [跟踪] 引入数据集跟踪 API： mlflow.data 和 mlflow.log_input() (#8186, @prithvikannan)
  • [跟踪] 添加 mlflow.log_table() 和 mlflow.load_table() APIs，用于记录评估表 (#8523, #8467, @sunishsheth2009)
  • [Tracking] 引入 mlflow.get_parent_run() 流畅的 API (#8493, @annzhang-db)
  • [跟踪 / 模型注册表] 在 Databricks 上重新引入更快的工件下载 (#8352, @dbczumar; #8561, @harupy)
  • [UI] 向 MLflow Tracking UI 添加数据集跟踪信息 (#8602, @prithvikannan, @hubertzub-db)
  • [UI] 引入 Artifact 视图，用于比较模型之间的输入、输出和元数据 (#8602, @hubertzub-db)
  • [Models] 将 mlflow.evaluate() 扩展以支持 LLM 任务 (#8484, @harupy)
  • [模型] 支持在 mlflow.langchain flavor 中记录 Chain 和 LLMChain 的子类 (#8453, @liangz1)
  • [模型] 向 mlflow.langchain flavor 添加对 LangChain 智能体 的支持 (#8297, @sunishsheth2009)
  • [模型] 为 SentenceTransformers 添加一个 mlflow.sentence_transformers flavor (#8479, @BenWilson2; #8547, @Loquats)
  • [模型] 添加对 mlflow.transformers flavor 的多GPU推理和高效权重加载的支持 (#8448, @ankit-db)
  • [模型] 在 mlflow.transformers flavor 中支持 max_shard_size 参数 (#8567, @wenfeiy-db)
  • [模型] 在 mlflow.transformers flavor 中添加对音频转录管道的支持 (#8464, @BenWilson2)
  • [模型] 为 mlflow.transformers flavor 添加对音频分类的支持 (#8492, @BenWilson2)
  • [Models] 在使用 mlflow.transformers flavor 记录的音频模型中添加对 URI 输入的支持 (#8495, @BenWilson2)
  • [Models] 添加对在 mlflow.transformers pyfunc 输出中返回分类器分数的支持 (#8512, @BenWilson2)
  • [模型] 支持模型签名中的可选输入 (#8438, @apurva-koti)
  • [模型] 引入一个 mlflow.models.set_signature() API 来设置已记录模型的签名 (#8476, @jerrylian-db)
  • [模型] 在使用 mlflow.onnx.log_model() 记录模型时保留 ONNX Runtime InferenceSession 选项 (#8433, @leqiao-1)

错误修复：

  • [Tracking] 在禁用 Spark Autologging 或 Spark Session 关闭时终止 Spark 回调服务器 (#8508, @WeichenXu123)
  • [Tracking] 修复 mlflow server 与 Flask<2.0 的兼容性 (#8463, @kevingreer)
  • [模型] 在批量推理期间将 mlflow.transformers pyfunc 的标量字符串输出转换为字符串列表 (#8546, @BenWilson2)
  • [Models] 修复了一个错误，该错误导致通过 mlflow models build-docker 安装过时的 pyenv 版本（#8488, @Hellzed）
  • [模型注册表] 当模型版本被删除时，从存储中移除别名 (#8459, @arpitjasa-db)

Documentation updates:

  • [文档] 发布一个用于模型选择和部署的新 MLOps 快速入门 (#8462, @lobrien)
  • [文档] 将 MLflavors 库添加到 Community Model Flavors 文档 (#8420, @benjaminbluhm)
  • [文档] 为 Registered Model Aliases 添加文档 (#8445, @arpitjasa-db)
  • [Docs] 修复文档中记录的 mlflow models CLI 命令示例中的错误 (#8480, @vijethmoudgalya)

有关完整的更改列表，请参阅 release change log，并查看最新的文档，见 mlflow.org。