Skip to content

Commit cd54e70

Browse files
authored
Fix non auto device map (#1005)
1 parent ead6f29 commit cd54e70

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

auto_round/utils/device.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -623,6 +623,8 @@ def set_non_auto_device_map(
623623
infos = device_map.split(",")
624624
device_map_dict = {}
625625
for info in infos:
626+
if ":" not in info:
627+
continue
626628
index = info.find(":")
627629
key = info[:index]
628630
value = info[index + 1 :]

0 commit comments

Comments
 (0)