Unverified Commit 8286b1d9 authored by Anjor Kanekar's avatar Anjor Kanekar Committed by GitHub
Browse files
parent 5503b274
...@@ -142,9 +142,7 @@ class HFLM(LM): ...@@ -142,9 +142,7 @@ class HFLM(LM):
+ [f"cuda:{i}" for i in range(torch.cuda.device_count())] + [f"cuda:{i}" for i in range(torch.cuda.device_count())]
+ ["mps", "mps:0"] + ["mps", "mps:0"]
) )
if device: if device and device in device_list:
if device not in device_list:
device = int(device)
self._device = torch.device(device) self._device = torch.device(device)
eval_logger.info(f"Using device '{device}'") eval_logger.info(f"Using device '{device}'")
if device in ("mps", "mps:0") and version.parse( if device in ("mps", "mps:0") and version.parse(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment