fix issue #7: checkpoint key

EMCarrami · May 19, 2024 · 57f7770 · 57f7770
1 parent 2bd8495
commit 57f7770
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pika/main.py b/pika/main.py
@@ -38,6 +38,7 @@ def __init__(self, config: Dict[str, Any], inference_only: bool = False) -> None
             config["model"].update(dict(self.model.hparams))
             self.model_state = "pretrained"
         else:
+            config["model"].pop("checkpoint", None)
             self.model = PikaModel(**config["model"])
             self.model_state = "init"