taco/glados-ladosp-tts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix latent bugs
Some checks are pending
Build and Publish Docker Images / build-cpu (push) Waiting to run
Details
Build and Publish Docker Images / build-cuda (push) Successful in 6m15s
Details
Build and Publish Docker Images / build-rocm (push) Successful in 6m48s
Details

Browse Source
This commit is contained in:
Stephen Tafoya
2026-06-13 02:20:24 -06:00
parent 89a6e73500
commit 4ad64ad1a8
2 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
wyoming_glados/handler.py
View File

@@ -41,11 +41,9 @@ def _detect_language(text: str) -> Languages:
def _load_bert_for_language(language: Languages, device: str) -> None:
model_name = _BERT_MODEL_NAMES[language]
if not bert_models.is_model_loaded(language):
_LOGGER.info("Loading BERT model for %s (%s)", language.name, model_name)
bert_models.load_model(language, model_name)
if not bert_models.is_tokenizer_loaded(language):
bert_models.load_tokenizer(language, model_name)
_LOGGER.info("Loading BERT model for %s (%s)", language.name, model_name)
bert_models.load_model(language, model_name)
bert_models.load_tokenizer(language, model_name)
bert = bert_models.__loaded_models.get(language)
if bert is not None:
bert = bert.float()