👍️[Update] Main/Controller : CTranslate2/Whisper model weight download のDownload済みの場合の処理を追加

This commit is contained in:
misyaguziya
2024-08-11 19:02:55 +09:00
parent 633cdf246c
commit 83bf21c20e
3 changed files with 9 additions and 7 deletions

View File

@@ -106,8 +106,8 @@ class Model:
def changeTranslatorCTranslate2Model(self):
self.translator.changeCTranslate2Model(config.PATH_LOCAL, config.CTRANSLATE2_WEIGHT_TYPE)
def downloadCTranslate2ModelWeight(self, callback=None):
return downloadCTranslate2Weight(config.PATH_LOCAL, config.CTRANSLATE2_WEIGHT_TYPE, callback)
def downloadCTranslate2ModelWeight(self, callbackFunc=None):
return downloadCTranslate2Weight(config.PATH_LOCAL, config.CTRANSLATE2_WEIGHT_TYPE, callbackFunc)
def isLoadedCTranslate2Model(self):
return self.translator.isLoadedCTranslate2Model()
@@ -115,8 +115,8 @@ class Model:
def checkTranscriptionWhisperModelWeight(self):
return checkWhisperWeight(config.PATH_LOCAL, config.WHISPER_WEIGHT_TYPE)
def downloadWhisperModelWeight(self, callback=None):
return downloadWhisperWeight(config.PATH_LOCAL, config.WHISPER_WEIGHT_TYPE, callback)
def downloadWhisperModelWeight(self, callbackFunc=None):
return downloadWhisperWeight(config.PATH_LOCAL, config.WHISPER_WEIGHT_TYPE, callbackFunc)
def resetKeywordProcessor(self):
del self.keyword_processor