feat: 增加推理部分工厂模式

2026-03-30 00:55:15 +08:00
parent 980299cd54
commit c01791ff54
17 changed files with 227 additions and 559 deletions
@@ -72,9 +72,12 @@ class BaseModelIO:
 class ModelParameter(BaseModelIO):
    """Container for model parameters with serialization capabilities."""
    
-    def save(self, save_dir: Union[str, Path]):
-        self.save_components(save_dir)
+    @classmethod
+    def save(cls, instance: "ModelParameter", save_dir: Union[str, Path]):
+        instance.save_components(save_dir)
    
-    def load(self, load_dir: Union[str, Path]) -> "ModelParameter":
-        return self.load_components(load_dir)
+    @classmethod
+    def load(cls, load_dir: Union[str, Path]) -> "ModelParameter":
+        instance = cls()
+        return instance.load_components(load_dir)