fix wandb secret

modal-labs · May 23, 2024 · 2925f41 · 2925f41
1 parent 921f6b4
commit 2925f41
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/common.py b/src/common.py
@@ -28,7 +28,6 @@
             HUGGINGFACE_HUB_CACHE="/pretrained",
             HF_HUB_ENABLE_HF_TRANSFER="1",
             TQDM_DISABLE="true",
-            ALLOW_WANDB=str(ALLOW_WANDB),
         )
     )
 )
@@ -42,8 +41,10 @@
 
 app = modal.App(
     APP_NAME,
-    secrets=[modal.Secret.from_name("huggingface")]
-    + ([modal.Secret.from_name("wandb")] if ALLOW_WANDB else []),
+    secrets=[
+        modal.Secret.from_name("huggingface"), 
+        (modal.Secret.from_name("wandb") if ALLOW_WANDB else modal.Secret.from_dict({})),
+    ],
 )
 
 # Volumes for pre-trained models and training runs.