apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-21 14:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add config for text to image skill

Browse source
This commit is contained in:
shenchucheng 2024-02-08 14:35:11 +08:00
parent 57befdac30
commit 02d78c3a01
3 changed files with 24 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/learn/text_to_image.py
View file

@ -26,9 +26,9 @@ async def text_to_image(text, size_type: str = "512x512", openai_api_key="", mod
"""
image_declaration = "data:image/png;base64,"
if CONFIG.METAGPT_TEXT_TO_IMAGE_MODEL_URL or model_url:
binary_data = await oas3_metagpt_text_to_image(text, size_type, model_url)
elif CONFIG.OPENAI_API_KEY or openai_api_key:
binary_data = await oas3_openai_text_to_image(text, size_type)
binary_data = await oas3_metagpt_text_to_image(text, size_type, model_url, **kwargs)
elif CONFIG.OPENAI_TEXT_TO_IMAGE_API_KEY or openai_api_key:
binary_data = await oas3_openai_text_to_image(text, size_type, openai_api_key, **kwargs)
else:
raise ValueError("Missing necessary parameters.")
base64_data = base64.b64encode(binary_data).decode("utf-8")