From e89d9723c5ec158ec7d38da394566e22d062186e Mon Sep 17 00:00:00 2001 From: Sebastian Raschka Date: Thu, 9 Jan 2025 09:59:03 -0600 Subject: [PATCH] Update litgpt/scripts/download.py Thanks! Co-authored-by: Andrei-Aksionov <58434077+Andrei-Aksionov@users.noreply.github.com> --- litgpt/scripts/download.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litgpt/scripts/download.py b/litgpt/scripts/download.py index ae04ecaa47..84fc7ee303 100644 --- a/litgpt/scripts/download.py +++ b/litgpt/scripts/download.py @@ -86,7 +86,7 @@ def download_from_hub( free_space_gb = free_space_bytes / (1024**3) if weight_size_gb > free_space_gb: - if os.getenv("LIGHTNING_CLUSTER_ID") is not None: + if os.getenv("LIGHTNING_CLOUD_SPACE_ID") is not None: studio_text = " Please switch to a larger Studio with more disk space." else: studio_text = ""