fix loading from non col checkpoint

webis-de · Aug 7, 2024 · e17fcf2 · e17fcf2
1 parent b8c5a8a
commit e17fcf2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lightning_ir/models/col/model.py b/lightning_ir/models/col/model.py
@@ -26,8 +26,7 @@ def from_pretrained(cls, model_name_or_path: str | Path, *args, **kwargs) -> Lig
             hf_hub_download(repo_id=str(model_name_or_path), filename="artifact.metadata")
         except Exception:
             return super().from_pretrained(model_name_or_path, *args, **kwargs)
-        finally:
-            return cls.from_colbert_checkpoint(model_name_or_path)
+        return cls.from_colbert_checkpoint(model_name_or_path)
 
     @classmethod
     def from_colbert_checkpoint(cls, model_name_or_path: Path | str) -> "ColModel":