Skip to content

Commit

Permalink
Merge pull request #200 from agwosdz/bug_ModelChoice
Browse files Browse the repository at this point in the history
Corrected Model choice
  • Loading branch information
rakuri255 authored Dec 21, 2024
2 parents 88a8dcc + f7e12b5 commit 38f5505
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/UltraSinger.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@
ultrastar_writer,
)
from modules.Speech_Recognition.TranscribedData import TranscribedData
from modules.Speech_Recognition.Whisper import WhisperModel
from modules.Ultrastar.ultrastar_score_calculator import Score, calculate_score_points
from modules.Ultrastar.ultrastar_txt import FILE_ENCODING, FormatVersion
from modules.Ultrastar.coverter.ultrastar_txt_converter import from_ultrastar_txt, \
Expand Down Expand Up @@ -602,7 +603,7 @@ def init_settings(argv: list[str]) -> Settings:
settings.output_folder_path = arg
elif opt in ("--whisper"):
settings.transcriber = "whisper"
settings.whisper_model = arg
settings.whisper_model = WhisperModel(arg)
elif opt in ("--whisper_align_model"):
settings.whisper_align_model = arg
elif opt in ("--whisper_batch_size"):
Expand Down

0 comments on commit 38f5505

Please sign in to comment.