SmerkyG jirka642 commited on
Commit
058a551
1 Parent(s): ab3f8be

Added missing super().__init__() in RWKV6State (#5)

Browse files

- Added missing super().__init__() in RWKV6State (762f342c01f55f2e9e25347e0eaf748d4d9c777f)


Co-authored-by: Jiří Kuneš <[email protected]>

Files changed (1) hide show
  1. modeling_rwkv6qwen2.py +1 -0
modeling_rwkv6qwen2.py CHANGED
@@ -60,6 +60,7 @@ _CONFIG_FOR_DOC = "RWKV6Qwen2Config"
60
 
61
  class RWKV6State(Cache):
62
  def __init__(self) -> None:
 
63
  self._seen_tokens = 0 # Used in `generate` to keep tally of how many tokens the cache has seen
64
  self.layer_kv_states: List[torch.Tensor] = []
65
  self.layer_shift_states: List[torch.Tensor] = []
 
60
 
61
  class RWKV6State(Cache):
62
  def __init__(self) -> None:
63
+ super().__init__()
64
  self._seen_tokens = 0 # Used in `generate` to keep tally of how many tokens the cache has seen
65
  self.layer_kv_states: List[torch.Tensor] = []
66
  self.layer_shift_states: List[torch.Tensor] = []