Skip to content

Commit

Permalink
models: add gpt4-turbo (#672)
Browse files Browse the repository at this point in the history
Signed-off-by: David Korczynski <[email protected]>
  • Loading branch information
DavidKorczynski authored Oct 22, 2024
1 parent 5df7682 commit d720ca6
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions llm_toolkit/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -290,23 +290,29 @@ def query_llm(self, prompt: prompts.Prompt, response_dir: str) -> None:


class GPT4(GPT):
"""OpenAI's GPTi-4 model."""
"""OpenAI's GPT-4 model."""

name = 'gpt-4'


class GPT4o(GPT):
"""OpenAI's GPTi-4o model."""
"""OpenAI's GPT-4o model."""

name = 'gpt-4o'


class GPT4oMini(GPT):
"""OpenAI's GPTi-4o-mini model."""
"""OpenAI's GPT-4o-mini model."""

name = 'gpt-4o-mini'


class GPT4Turbo(GPT):
"""OpenAI's GPT-4 Turbo model."""

name = 'gpt-4-turbo'


class AzureGPT(GPT):
"""Azure's GPT model."""

Expand Down

0 comments on commit d720ca6

Please sign in to comment.