OpenAIImageGeneration

运行异步 API 客户端的 OpenAI 图像生成实现。

属性

• model: 用于 ImageGenerationModel 的模型名称，例如 "dall-e-3" 等。支持的模型可以在这里找到。

• base_url: 用于 OpenAI API 请求的基础 URL。默认为 None，这意味着将使用为环境变量 OPENAI_BASE_URL 设置的值，如果未设置，则使用 "https://api.openai.com/v1"。

• api_key: 用于验证对 OpenAI API 请求的 API 密钥。默认为 None，这意味着将使用为环境变量 OPENAI_API_KEY 设置的值，如果未设置，则为 None。

• max_retries: 在失败之前，重试 API 请求的最大次数。默认为 6。

• timeout: 等待 API 响应的最长时间（秒）。默认为 120。

示例

从文本提示生成图像

from distilabel.models.image_generation import OpenAIImageGeneration

igm = OpenAIImageGeneration(model="dall-e-3", api_key="api.key")

igm.load()

output = igm.generate_outputs(
    inputs=["a white siamese cat"],
    size="1024x1024",
    quality="standard",
    style="natural",
)
# [{"images": ["iVBORw0KGgoAAAANSUhEUgA..."]}]