Added missing super().__init__() in RWKV6State (#5)
Browse files- Added missing super().__init__() in RWKV6State (762f342c01f55f2e9e25347e0eaf748d4d9c777f)
Co-authored-by: Jiří Kuneš <[email protected]>
- 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] = []
|