chore: 更新脚本并且修改gitignore

2026-04-02 15:40:31 +08:00
parent 475de51c7d
commit 912d7c7f54
6 changed files with 17 additions and 12 deletions
@@ -81,7 +81,7 @@ class BaseModelIO:
        self.config.load(str(paths["config"]))
        self.tokenizer.load(str(paths["tokenizer"]))

-        if self.model is None:
+        if isinstance(self.model, nn.Identity):
            with disable_random_init(enable=disable_init):
                self.model = Transformer(self.config)