From fb6708b55535b16c343ef8eb0cb4fdadbafae5ff Mon Sep 17 00:00:00 2001 From: LoganDark Date: Sat, 3 Jun 2023 03:09:51 -0700 Subject: [PATCH] Fix pytorch storage warnings, fixes #80 (#88) we seriously don't care what type of storage we get, pytorch sucks --- rwkv/rwkv_cpp_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/rwkv/rwkv_cpp_model.py b/rwkv/rwkv_cpp_model.py index dd195ef..d957de0 100644 --- a/rwkv/rwkv_cpp_model.py +++ b/rwkv/rwkv_cpp_model.py @@ -84,7 +84,7 @@ class RWKVModel: if state_in is not None: validate_buffer(state_in, 'state_in', self._state_buffer_element_count) - state_in_ptr = state_in.untyped_storage().data_ptr() + state_in_ptr = state_in.data_ptr() else: state_in_ptr = 0 @@ -102,8 +102,8 @@ class RWKVModel: self._ctx, token, state_in_ptr, - state_out.untyped_storage().data_ptr(), - logits_out.untyped_storage().data_ptr() + state_out.data_ptr(), + logits_out.data_ptr() ) return logits_out, state_out