diff --git a/src/forge/env.py b/src/forge/env.py index b698b8013..989cacba5 100644 --- a/src/forge/env.py +++ b/src/forge/env.py @@ -105,6 +105,17 @@ def get_value(self) -> Any: description="Whether or not to use RDMA in TorchStore.", ) +MONARCH_OLD_ASYNC_WORKAROUND = EnvVar( + name="MONARCH_OLD_ASYNC_WORKAROUND", + default=1, + description=( + "If enabled, monarch messages will be sent immediately even it's not" + " awaited. This is needed for parallel fetching of weights, as using" + " create_task creates race condition. This is a temporary workaround" + " and will be removed once we have a better solution." + ), +) + def all_env_vars() -> list[EnvVar]: """Retrieves all registered environment variable names."""