whisper模型改为medium

d243be95 · Administrator · e114b418 · d243be95
Commit d243be95 authored Nov 14, 2023 by Administrator
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

load_args.py app/video_cut/load_args.py +2 -1

No files found.
--- a/app/video_cut/load_args.py
+++ b/app/video_cut/load_args.py
@@ -134,7 +134,8 @@ def main_args(logger, debug):
    args.vad = "0"

    if not debug:
-        args.whisper_model = "large-v2"
+        # args.whisper_model = "large-v2"
+        args.whisper_model = "medium"
        args.device = "cuda"

    logger.info(f'load whisper_model: {args.whisper_model}, device: {args.device}')