mlflow.groq

mlflow.groq 模块提供了用于记录和加载 Groq 模型的 API。

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

注意

Autologging 已知与以下包版本兼容:0.13.1 <= groq <= 0.37.1。当使用此范围之外的包版本时,Autologging 可能不会成功。

启用(或禁用)并配置从 Groq 到 MLflow 的 autologging。仅支持同步调用。不会记录异步 API 和流。

参数
  • log_traces – 如果为 True,则为 Groq 模型记录跟踪。如果为 False,则在推理过程中不收集任何跟踪。默认为 True

  • disable – 如果为 True,则禁用 Groq autologging。默认为 False

  • silent – 如果为 True,则在 Groq autologging 期间抑制 MLflow 的所有事件日志和警告。如果为 False,则显示所有事件和警告。