Spaces:

howard-hou
/

VisualRWKV-Gradio-1

Sleeping

howard-hou commited on Dec 30, 2023

Commit

934f392

•

1 Parent(s): aa4c647

Update modeling_rwkv.py

Files changed (1) hide show

modeling_rwkv.py CHANGED Viewed

@@ -1032,6 +1032,7 @@ class RWKV(MyModule):
                 x = w['emb.weight'][tokens if seq_mode else tokens[0]]
             else:
                 x = embs
             for i in range(args.n_layer):
                 bbb = f'blocks.{i}.'

                 x = w['emb.weight'][tokens if seq_mode else tokens[0]]
             else:
                 x = embs
+                seq_mode = True
             for i in range(args.n_layer):
                 bbb = f'blocks.{i}.'