tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix logging for mps in auto_select_torch_device (#1513)

Browse Source
This commit is contained in:
Eugene Mironov
2025-07-16 02:23:00 +07:00
committed by GitHub
parent 1c0ac8e341
commit 3efb4410f1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/lerobot/utils/utils.py
View File

@@ -48,7 +48,7 @@ def auto_select_torch_device() -> torch.device:
logging.info("Cuda backend detected, using cuda.") logging.info("Cuda backend detected, using cuda.")
return torch.device("cuda") return torch.device("cuda")
elif torch.backends.mps.is_available(): elif torch.backends.mps.is_available():
logging.info("Metal backend detected, using cuda.") logging.info("Metal backend detected, using mps.")
return torch.device("mps") return torch.device("mps")
else: else:
logging.warning("No accelerated backend detected. Using default cpu, this will be slow.") logging.warning("No accelerated backend detected. Using default cpu, this will be slow.")