From 23cbe8c4b990baaee9d85773010e32fa95819730 Mon Sep 17 00:00:00 2001 From: rrutmann Date: Tue, 16 Sep 2025 14:06:47 +0200 Subject: [PATCH] feat: Resolve LOCAL_WORLD_SIZE to integer --- src/modalities/config/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/modalities/config/config.py b/src/modalities/config/config.py index 1365fb637..0ed8785e9 100644 --- a/src/modalities/config/config.py +++ b/src/modalities/config/config.py @@ -503,7 +503,7 @@ def load_app_config_dict( """ def cuda_env_resolver_fun(var_name: str) -> int: - int_env_variable_names = ["LOCAL_RANK", "WORLD_SIZE", "RANK"] + int_env_variable_names = ["LOCAL_RANK", "WORLD_SIZE", "RANK", "LOCAL_WORLD_SIZE"] return int(os.getenv(var_name)) if var_name in int_env_variable_names else os.getenv(var_name) def modalities_env_resolver_fun(var_name: str, kwargs: dict[str, Any]) -> str | Path: