ed
/
rwkv.cpp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Silence PyTorch warnings by using untyped storage (#72)

This commit is contained in:
LoganDark 2023-05-26 05:21:18 -07:00 committed by GitHub
parent 7cbfbc55c8
commit d26791b5bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
rwkv/rwkv_cpp_model.py
View File

@ -79,7 +79,7 @@ class RWKVModel:
if state_in is not None: if state_in is not None:
validate_buffer(state_in, 'state_in', self._state_buffer_element_count) validate_buffer(state_in, 'state_in', self._state_buffer_element_count)
state_in_ptr = state_in.storage().data_ptr() state_in_ptr = state_in.untyped_storage().data_ptr()
else: else:
state_in_ptr = 0 state_in_ptr = 0
@ -97,8 +97,8 @@ class RWKVModel:
self._ctx, self._ctx,
token, token,
state_in_ptr, state_in_ptr,
state_out.storage().data_ptr(), state_out.untyped_storage().data_ptr(),
logits_out.storage().data_ptr() logits_out.untyped_storage().data_ptr()
) )
return logits_out, state_out return logits_out, state_out