diff --git a/src/common.py b/src/common.py index 21b3df0..45f2fee 100644 --- 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.