From e620d171bcd871233b4fa08dccf5e2720fe9ca13 Mon Sep 17 00:00:00 2001
From: u010280923 <xinlu.nlp@foxmail.com>
Date: Mon, 20 Mar 2023 16:25:50 +0800
Subject: [PATCH] bug fixed

---
 src/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/model.py b/src/model.py
index cae2594..ee5169d 100644
--- 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
-- 
GitLab