Update litgpt/scripts/download.py

Thanks! Co-authored-by: Andrei-Aksionov <58434077+Andrei-Aksionov@users.noreply.github.com>
Lightning-AI · Jan 9, 2025 · e89d972 · e89d972
1 parent 9f563d3
commit e89d972
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 = ""