[QA Bug] Remove NGC dependency (#494)

This allows the checkpoint to be optionally downloaded from PBSS for QA testing.
NVIDIA · Dec 3, 2024 · 7f9fd66 · 7f9fd66
1 parent 793f407
commit 7f9fd66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sub-packages/bionemo-esm2/src/bionemo/esm2/model/finetune/train.py b/sub-packages/bionemo-esm2/src/bionemo/esm2/model/finetune/train.py
@@ -175,7 +175,7 @@ def train_model(
     seed = 42
 
     # To download a 650M pre-trained ESM2 model
-    pretrain_ckpt_path = load("esm2/650m:2.0", source="ngc")
+    pretrain_ckpt_path = load("esm2/650m:2.0")
 
     config = ESM2FineTuneSeqConfig(initial_ckpt_path=str(pretrain_ckpt_path))