Change faster-whisper as default

This commit is contained in:
litagin02
2024-11-11 10:25:30 +09:00
parent 038d07f4e7
commit 875186f0c7

View File

@@ -174,12 +174,13 @@ def create_dataset_app() -> gr.Blocks:
) )
use_hf_whisper = gr.Checkbox( use_hf_whisper = gr.Checkbox(
label="HuggingFaceのWhisperを使う速度が速いがVRAMを多く使う", label="HuggingFaceのWhisperを使う速度が速いがVRAMを多く使う",
value=True, value=False,
) )
hf_repo_id = gr.Dropdown( hf_repo_id = gr.Dropdown(
["openai/whisper", "kotoba-tech/kotoba-whisper-v1.1"], ["openai/whisper", "kotoba-tech/kotoba-whisper-v1.1"],
label="HuggingFaceのWhisperモデル", label="HuggingFaceのWhisperモデル",
value="openai/whisper", value="openai/whisper",
visible=False,
) )
compute_type = gr.Dropdown( compute_type = gr.Dropdown(
[ [
@@ -194,7 +195,7 @@ def create_dataset_app() -> gr.Blocks:
], ],
label="計算精度", label="計算精度",
value="bfloat16", value="bfloat16",
visible=False, visible=True,
) )
batch_size = gr.Slider( batch_size = gr.Slider(
minimum=1, minimum=1,
@@ -203,6 +204,7 @@ def create_dataset_app() -> gr.Blocks:
step=1, step=1,
label="バッチサイズ", label="バッチサイズ",
info="大きくすると速度が速くなるがVRAMを多く使う", info="大きくすると速度が速くなるがVRAMを多く使う",
visible=False,
) )
language = gr.Dropdown(["ja", "en", "zh"], value="ja", label="言語") language = gr.Dropdown(["ja", "en", "zh"], value="ja", label="言語")
initial_prompt = gr.Textbox( initial_prompt = gr.Textbox(