File "/data2/yzw/001/video/slowfast/models/Uvrwkv6.py", line 250, in _inner_forward
x = RUN_CUDA_RWKV6(B, T, C, self.n_head, r, k, v, w, u=self.time_faaaa)
File "/data2/yzw/001/video/slowfast/models/Uvrwkv6.py", line 89, in RUN_CUDA_RWKV6
return WKV_6.apply(B, T, C, H, r, k, v, w, u)
File "/data2/yzw/miniconda3/lib/python3.12/site-packages/torch/autograd/function.py", line 574, in apply
return super().apply(*args, **kwargs) # type: ignore[misc]
RuntimeError: _Map_base::at
作者您好，在我使用vision rwkv处理任务时遭遇了这个问题，您有时间的话帮忙解答一下
File "/data2/yzw/001/video/slowfast/models/Uvrwkv6.py", line 250, in _inner_forward
x = RUN_CUDA_RWKV6(B, T, C, self.n_head, r, k, v, w, u=self.time_faaaa)
File "/data2/yzw/001/video/slowfast/models/Uvrwkv6.py", line 89, in RUN_CUDA_RWKV6
return WKV_6.apply(B, T, C, H, r, k, v, w, u)
File "/data2/yzw/miniconda3/lib/python3.12/site-packages/torch/autograd/function.py", line 574, in apply
return super().apply(*args, **kwargs) # type: ignore[misc]
RuntimeError: _Map_base::at
作者您好，在我使用vision rwkv处理任务时遭遇了这个问题，您有时间的话帮忙解答一下