Fix logging for mps in auto_select_torch_device (#1513)

2025-07-16 02:23:00 +07:00
parent 1c0ac8e341
commit 3efb4410f1
1 changed files with 1 additions and 1 deletions
--- a/src/lerobot/utils/utils.py
+++ b/src/lerobot/utils/utils.py
@@ -48,7 +48,7 @@ def auto_select_torch_device() -> torch.device:
        logging.info("Cuda backend detected, using cuda.")
        return torch.device("cuda")
    elif torch.backends.mps.is_available():
-        logging.info("Metal backend detected, using cuda.")
+        logging.info("Metal backend detected, using mps.")
        return torch.device("mps")
    else:
        logging.warning("No accelerated backend detected. Using default cpu, this will be slow.")