From f48deab320e79bfb56da866d10f66c561eb8bb71 Mon Sep 17 00:00:00 2001 From: Rene Teigen Date: Tue, 31 Jan 2023 10:01:49 +0000 Subject: [PATCH] Forgot to remove extra models --- models/openai_model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/models/openai_model.py b/models/openai_model.py index 45e97b4..ead92ca 100644 --- a/models/openai_model.py +++ b/models/openai_model.py @@ -45,7 +45,6 @@ class Models: # Text models DAVINCI = "text-davinci-003" CURIE = "text-curie-001" - BABBAGES = "babbage" BABBAGE = "text-babbage-001" ADA = "text-ada-001" @@ -61,7 +60,7 @@ class Models: CODE_EDIT = "code-davinci-edit-001" # Model collections - TEXT_MODELS = [DAVINCI, CURIE, BABBAGES, BABBAGE, ADA, CODE_DAVINCI, CODE_CUSHMAN] + TEXT_MODELS = [DAVINCI, CURIE, BABBAGE, ADA, CODE_DAVINCI, CODE_CUSHMAN] EDIT_MODELS = [EDIT, CODE_EDIT] DEFAULT = DAVINCI