🚧[WIP/TEST] Model : 翻訳起動時にモデルをロードするように変更

This commit is contained in:
misyaguziya
2024-02-14 00:21:56 +09:00
parent bc0d2f246b
commit 4eb811aac4
3 changed files with 13 additions and 2 deletions

View File

@@ -1,3 +1,4 @@
import gc
import os
from deepl import Translator as deepl_Translator
from translators import translate_text as other_web_Translator
@@ -44,6 +45,13 @@ class Translator():
tokenizer_path = os.path.join("./weights", "ctranslate2", directory_name, "tokenizer")
self.ctranslate2_tokenizer = transformers.AutoTokenizer.from_pretrained(tokenizer, cache_dir=tokenizer_path)
def clearCTranslate2Model(self):
del self.ctranslate2_translator
del self.ctranslate2_tokenizer
gc.collect()
self.ctranslate2_translator = None
self.ctranslate2_tokenizer = None
@staticmethod
def getLanguageCode(translator_name, target_country, source_language, target_language):
match translator_name: