We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c2a2625 commit 04a9dc3Copy full SHA for 04a9dc3
1 file changed
FlagEmbedding/abc/inference/AbsReranker.py
@@ -96,7 +96,7 @@ def get_target_devices(devices: Union[str, int, List[str], List[int]]) -> List[s
96
elif is_torch_npu_available():
97
return [f"npu:{i}" for i in range(torch.npu.device_count())]
98
elif torch.backends.mps.is_available():
99
- return [f"mps:{i}" for i in range(torch.mps.device_count())]
+ return ["mps"]
100
else:
101
return ["cpu"]
102
elif isinstance(devices, str):
0 commit comments