Skip to content

Commit bfb030d

Browse files
authored
Bugfix of modelarts (#671)
1 parent 5b28da0 commit bfb030d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/modelarts_adapter/modelarts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def run_with_single_rank(local_rank: int = 0, signal: str = "/tmp/SUCCESS") -> C
6767
def decorator(func: Callable[..., Any]) -> Callable[..., Any]:
6868
@functools.wraps(func)
6969
def wrapper(*args: Any, **kwargs: Any) -> Any:
70-
if local_rank in [0, None]:
70+
if local_rank is None or local_rank % 8 == 0:
7171
result = func(*args, **kwargs)
7272
with open(signal, "w") as f:
7373
f.write("\n")

0 commit comments

Comments
 (0)