Skip to content

Commit

Permalink
fix: Aligned the flag naming convention with the argument naming conv…
Browse files Browse the repository at this point in the history
…ention.

Change-Id: I8a284d4cfdbd3f301b2c527b791ad77ed1619d0a
  • Loading branch information
Kacper Krasowiak committed Oct 25, 2024
1 parent 9eb260e commit b334766
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,8 +178,8 @@
40,
"Top-k sampling parameter.",
)
_GEMINI_SAFETY_SETTINGS = flags.DEFINE_string(
"gemini_safety_settings",
_SAFETY_SETTINGS = flags.DEFINE_string(
"_safety_settings",
"Medium",
"The indicator of what kind of Gemini safety settings should"
" be used in the dubbing process. Can be"
Expand Down Expand Up @@ -253,7 +253,7 @@ def main(argv: Sequence[str]) -> None:
top_p=_TOP_P.value,
top_k=_TOP_K.value,
max_output_tokens=_MAX_OUTPUT_TOKENS.value,
safety_settings=get_safety_settings(_GEMINI_SAFETY_SETTINGS.value),
safety_settings=get_safety_settings(_SAFETY_SETTINGS.value),
use_elevenlabs=False if _ELEVENLABS_TOKEN.value == "".strip() else True,
elevenlabs_token=_ELEVENLABS_TOKEN.value,
elevenlabs_clone_voices=_ELEVENLABS_CLONE_VOICES.value,
Expand Down

0 comments on commit b334766

Please sign in to comment.