theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rwkv.cpp python38 compatibility

Browse Source
This commit is contained in:
josc146 2023-12-12 23:19:18 +08:00
parent 26b30f0dbe
commit 82c9825da8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend-python/rwkv_pip/cpp/model.py vendored
View File

@ -1,4 +1,4 @@
from typing import Any, List
from typing import Any, List, Union
from . import rwkv_cpp_model
from . import rwkv_cpp_shared_library
@ -10,5 +10,5 @@ class RWKV:
self.w = {} # fake weight
self.w["emb.weight"] = [0] * self.model.n_vocab
def forward(self, tokens: List[int], state: Any | None):
def forward(self, tokens: List[int], state: Union[Any, None] = None):
return self.model.eval_sequence_in_chunks(tokens, state, use_numpy=True)