fix rocm hip mismatch to cuda

2026-06-13 02:33:57 -06:00
parent 4ad64ad1a8
commit a0ff64629c
2 changed files with 3 additions and 2 deletions
--- a/wyoming_glados/handler.py
+++ b/wyoming_glados/handler.py
@@ -48,8 +48,9 @@ def _load_bert_for_language(language: Languages, device: str) -> None:
    if bert is not None:
        bert = bert.float()
        bert.eval()
+        bert.to(device)
        bert_models.__loaded_models[language] = bert
-        _LOGGER.info("BERT model for %s cast to float32", language.name)
+        _LOGGER.info("BERT model for %s cast to float32 and moved to %s", language.name, device)


 def _find_model_files(model_dir: Path):