mlflow.mistral

mlflow.mistral.autolog(log_traces: bool = True, disable: bool = False, silent: bool = False)[source]

注意

实验性:此功能可能在将来的发布中更改或在不另行通知的情况下被移除。

注意

已知 Autologging 与以下包版本兼容: 1.0.0 <= mistralai <= 1.9.3。在与超出此范围的包版本一起使用时,Autologging 可能无法成功。

启用(或禁用)并配置从 Mistral AI 到 MLflow 的自动日志记录。仅支持对 Text generation API 的同步调用。异步 API 和流式传输不会被记录。

Parameters
  • log_traces – 如果 True,将为 Mistral AI 模型记录跟踪信息。 如果 False,在推理期间不会收集跟踪信息。默认值为 True

  • disable – 如果 True,则禁用 Mistral AI 的自动日志记录。默认为 False

  • silent – 如果 True,在 Mistral AI 自动记录期间抑制 MLflow 的所有事件日志和警告。如果 False,则显示所有事件和警告。