Apply a more defensive way to fix redownloading issue (#55)

Author: james77777778

kimm/__init__.py

@@ -13,4 +13,4 @@
 from kimm._src.utils.model_registry import list_models
 from kimm._src.version import version
 
-__version__ = "0.2.4"
+__version__ = "0.2.5"

kimm/_src/models/base_model.py

@@ -271,6 +271,14 @@ def get_config(self):
     def fix_config(self, config: typing.Dict):
         return config
 
+    @classmethod
+    def from_config(cls, config, custom_objects=None):
+        # Set `self._weight` to `None` to avoid redownloading issue.
+        config = config.copy()
+        if "weights" in config:
+            config["weights"] = None
+        return cls(**config)
+
     def __repr__(self):
         repr_str = (
             f"<{self.__class__.__name__} "

kimm/_src/version.py

@@ -1,6 +1,6 @@
 from kimm._src.kimm_export import kimm_export
 
-__version__ = "0.2.4"
+__version__ = "0.2.5"
 
 
 @kimm_export("kimm")