chao.wu/Lerobot
1
0
Fork 0
Code Pull Requests 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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.")