fix latent bugs

wyoming_glados/__main__.py

@@ -23,13 +23,18 @@ async def main() -> None:
     parser.add_argument("--model-dir", type=Path, required=True,
                         help="Directory containing model files (config.json, *.safetensors, style_vectors.npy)")
     parser.add_argument("--device", default="cpu",
-                        help="Device for PyTorch (cpu, cuda)")
+                        help="Device for PyTorch (cpu, cuda, rocm)")
     parser.add_argument("--debug", action="store_true",
                         help="Log DEBUG messages")
     parser.add_argument("--version", action="version",
                         version=__version__)
     args = parser.parse_args()
 
+    device_map = {
+        "rocm": "hip",
+    }
+    device = device_map.get(args.device, args.device)
+
     logging.basicConfig(
         level=logging.DEBUG if args.debug else logging.INFO,
         format="%(asctime)s %(levelname)s %(name)s %(message)s",
@@ -72,7 +77,7 @@ async def main() -> None:
 
     _LOGGER.info("Starting GLaDOS Wyoming TTS server on %s", args.uri)
     _LOGGER.info("Model directory: %s", model_dir)
-    _LOGGER.info("Device: %s", args.device)
+    _LOGGER.info("Device: %s", device)
 
     server_task = asyncio.create_task(
         server.run(
@@ -80,7 +85,7 @@ async def main() -> None:
                 GLaDOSEventHandler,
                 wyoming_info,
                 model_dir,
-                args.device,
+                device,
             )
         )
     )