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

fix rocm hip mismatch to cuda
All checks were successful
Build and Publish Docker Images / build-cuda (push) Successful in 4m52s
Details
Build and Publish Docker Images / build-rocm (push) Successful in 6m59s
Details
Build and Publish Docker Images / build-cpu (push) Successful in 24m31s
Details

Browse Source
This commit is contained in:
Stephen Tafoya
2026-06-13 02:33:57 -06:00
parent 4ad64ad1a8
commit a0ff64629c
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
wyoming_glados/__main__.py
View File

@@ -31,7 +31,7 @@ async def main() -> None:
args = parser.parse_args() args = parser.parse_args()
device_map = { device_map = {
"rocm": "hip", "rocm": "cuda",
} }
device = device_map.get(args.device, args.device) device = device_map.get(args.device, args.device)

3
wyoming_glados/handler.py
View File

@@ -48,8 +48,9 @@ def _load_bert_for_language(language: Languages, device: str) -> None:
if bert is not None: if bert is not None:
bert = bert.float() bert = bert.float()
bert.eval() bert.eval()
bert.to(device)
bert_models.__loaded_models[language] = bert 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): def _find_model_files(model_dir: Path):