From 3ac39851806f7aae2240686188e9bfd7f0bd0fd3 Mon Sep 17 00:00:00 2001 From: Mark Sze Date: Tue, 17 Dec 2024 01:08:43 +0000 Subject: [PATCH] Updated resolutions for dall-e-3 Signed-off-by: Mark Sze --- .../capabilities/test_image_generation_capability.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test/agentchat/contrib/capabilities/test_image_generation_capability.py b/test/agentchat/contrib/capabilities/test_image_generation_capability.py index abd73d52ce..c050a775af 100644 --- a/test/agentchat/contrib/capabilities/test_image_generation_capability.py +++ b/test/agentchat/contrib/capabilities/test_image_generation_capability.py @@ -34,11 +34,11 @@ filter_dict = {"model": ["gpt-4o-mini"]} -RESOLUTIONS = ["256x256", "512x512", "1024x1024"] +RESOLUTIONS = ["1024x1024", "1792x1024", "1024x1792"] QUALITIES = ["standard", "hd"] PROMPTS = [ - "Generate an image of a robot holding a 'I Love Autogen' sign", - "Generate an image of a dog holding a 'I Love Autogen' sign", + "Generate an image of a robot holding a 'I Love AG2' sign", + "Generate an image of a dog holding a 'I Love AG2' sign", ] @@ -228,5 +228,5 @@ def test_image_generation_capability_cache(monkeypatch): if __name__ == "__main__": test_dalle_image_generator( - dalle_config={"config_list": openai_utils.config_list_from_models(model_list=["dall-e-2"], exclude="aoai")} + dalle_config={"config_list": openai_utils.config_list_from_models(model_list=["dall-e-3"], exclude="aoai")} )