bug fixed

e620d171 · u010280923 · 02e8a2d9 · e620d171
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

src/model.py src/model.py +1 -1

未找到文件。
--- a/src/model.py
+++ b/src/model.py
@@ -385,7 +385,7 @@ class RWKV(pl.LightningModule):
    def load(self, path):
        path = Path(path)
        assert path.exists()
-        self.load_state_dict(torch.load(str(path)), map_location="cpu")
+        self.load_state_dict(torch.load(str(path), map_location="cpu"))

    def configure_optimizers(self):
        args = self.args