From d26791b5bc758bc712be3e4e209f02a024a782f5 Mon Sep 17 00:00:00 2001 From: LoganDark Date: Fri, 26 May 2023 05:21:18 -0700 Subject: [PATCH] Silence PyTorch warnings by using untyped storage (#72) --- 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 f7bb32b..247002d 100644 --- a/rwkv/rwkv_cpp_model.py +++ b/rwkv/rwkv_cpp_model.py @@ -79,7 +79,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.storage().data_ptr() + state_in_ptr = state_in.untyped_storage().data_ptr() else: state_in_ptr = 0 @@ -97,8 +97,8 @@ class RWKVModel: self._ctx, token, state_in_ptr, - state_out.storage().data_ptr(), - logits_out.storage().data_ptr() + state_out.untyped_storage().data_ptr(), + logits_out.untyped_storage().data_ptr() ) return logits_out, state_out