From 4fbf7bd067e510f4d845a0da937e570e3a135536 Mon Sep 17 00:00:00 2001 From: Ray Tien Date: Sun, 29 Sep 2024 15:48:11 +0800 Subject: [PATCH] Add gpt-4o-mini to MultiModel Engines. --- examples/notebooks/Tutorial-MultiModal.ipynb | 4 ++-- textgrad/engine/__init__.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/examples/notebooks/Tutorial-MultiModal.ipynb b/examples/notebooks/Tutorial-MultiModal.ipynb index f56e5b8..d88a1d5 100644 --- a/examples/notebooks/Tutorial-MultiModal.ipynb +++ b/examples/notebooks/Tutorial-MultiModal.ipynb @@ -61,7 +61,7 @@ "metadata": {}, "outputs": [], "source": [ - "tg.set_backward_engine(\"gpt-4o\")" + "tg.set_backward_engine(\"gpt-4o-mini\")" ] }, { @@ -138,7 +138,7 @@ ], "source": [ "question_variable = tg.Variable(\"What do you see in this image?\", role_description=\"question\", requires_grad=False)\n", - "response = MultimodalLLMCall(\"gpt-4o\")([image_variable, question_variable])\n", + "response = MultimodalLLMCall(\"gpt-4o-mini\")([image_variable, question_variable])\n", "response" ] }, diff --git a/textgrad/engine/__init__.py b/textgrad/engine/__init__.py index 3e0ee24..fe12215 100644 --- a/textgrad/engine/__init__.py +++ b/textgrad/engine/__init__.py @@ -12,6 +12,7 @@ # Any better way to do this? __MULTIMODAL_ENGINES__ = ["gpt-4-turbo", "gpt-4o", + "gpt-4o-mini", "claude-3-5-sonnet-20240620", "claude-3-opus-20240229", "claude-3-sonnet-20240229",