Skip to content

Commit b856127

Browse files
authored
Merge pull request #398 from Modalities/resolve_local_world_size
feat: Resolve LOCAL_WORLD_SIZE to integer
2 parents bf0e6d3 + fb9c5fd commit b856127

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/modalities/config/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -545,7 +545,7 @@ def load_app_config_dict(
545545
"""
546546

547547
def cuda_env_resolver_fun(var_name: str) -> int | str | None:
548-
int_env_variable_names = ["LOCAL_RANK", "WORLD_SIZE", "RANK"]
548+
int_env_variable_names = ["LOCAL_RANK", "WORLD_SIZE", "RANK", "LOCAL_WORLD_SIZE"]
549549
return int(os.environ[var_name]) if var_name in int_env_variable_names else os.getenv(var_name)
550550

551551
def modalities_env_resolver_fun(var_name: str, kwargs: dict[str, Any]) -> str | Path:

0 commit comments

Comments
 (0)