Spaces:

howard-hou
/

VisualRWKV-Gradio-1

Sleeping

howard-hou commited on Dec 30, 2023

Commit

aa4c647

•

1 Parent(s): c397d4d

Update modeling_rwkv.py

Files changed (1) hide show

modeling_rwkv.py CHANGED Viewed

@@ -1000,7 +1000,7 @@ class RWKV(MyModule):
     ########################################################################################################
-    def forward(self, tokens, state, full_output=False, embs=None):
         with torch.no_grad():
             w = self.w
             args = self.args

     ########################################################################################################
+    def forward(self, tokens=None, state=None, full_output=False, embs=None):
         with torch.no_grad():
             w = self.w
             args = self.args