Skip to content

Commit

Permalink
Change GPT default model to gpt-4o-mini
Browse files Browse the repository at this point in the history
  • Loading branch information
@ionic-bond
ionic-bond committed Oct 11, 2024
1 parent 96151e0 commit 9fa4ef2
Show file tree
Hide file tree
Showing 3 changed files with 46 additions and 46 deletions.
6 changes: 3 additions & 3 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,7 @@ python3 ./stream-translator-gpt/translator.py
| `--prefix_retention_length` | 0.5 | The length of the retention prefix audio during slicing. |
| `--vad_threshold` | 0.35 | The threshold of Voice activity detection. if the speech probability of a frame is higher than this value, then this frame is speech. |
| **Transcription Options** |
| `--model` | small | Select Whisper/Faster-Whisper model size. See [here](https://github.com/openai/whisper#available-models-and-languages) for available models. |
| `--model` | small | Select Whisper/Faster-Whisper model size. See [here](https://github.com/openai/whisper#available-models-and-languages) for available models. |
| `--language` | auto | Language spoken in the stream. See [here](https://github.com/openai/whisper#available-models-and-languages) for available languages. |
| `--beam_size` | 5 | Number of beams in beam search. Set to 0 to use greedy algorithm instead (faster but less accurate). |
| `--best_of` | 5 | Number of candidates when sampling with non-zero temperature. |
Expand All @@ -170,8 +170,8 @@ python3 ./stream-translator-gpt/translator.py
| **Translation Options** |
| `--openai_api_key` | | OpenAI API key if using GPT translation / Whisper API. |
| `--google_api_key` | | Google API key if using Gemini translation. |
| `--gpt_model` | gpt-3.5-turbo | OpenAI's GPT model name, gpt-3.5-turbo / gpt-4 / gpt-4o. |
| `--gemini_model` | gemini-1.0-pro | Google's Gemini model name, gemini-1.0-pro / gemini-1.5-flash-latest / gemini-1.5-pro-latest |
| `--gpt_model` | gpt-4o-mini | OpenAI's GPT model name, gpt-3.5-turbo / gpt-4o / gpt-4o-mini. |
| `--gemini_model` | gemini-1.0-pro | Google's Gemini model name, gemini-1.0-pro / gemini-1.5-flash / gemini-1.5-pro |
| `--gpt_translation_prompt` | | If set, will translate the result text to target language via GPT / Gemini API (According to which API key is filled in). Example: "Translate from Japanese to Chinese" |
| `--gpt_translation_history_size` | 0 | The number of previous messages sent when calling the GPT / Gemini API. If the history size is 0, the translation will be run parallelly. If the history size > 0, the translation will be run serially. |
| `--gpt_translation_timeout` | 10 | If the GPT / Gemini translation exceeds this number of seconds, the translation will be discarded. |
Expand Down
Loading

0 comments on commit 9fa4ef2

Please sign in to comment.