image_indices = (k_states_image_token_L1_norm.topk(pivot_image_token).indices + image_token_start_index).tolist()
RuntimeError: selected index k out of range
and the k_states_image_token_L1_norm is in shape of torch.Size([0]), and k_states_image_token'shape is torch.Size([0, 512])
image_indices = (k_states_image_token_L1_norm.topk(pivot_image_token).indices + image_token_start_index).tolist()RuntimeError: selected index k out of range
and the k_states_image_token_L1_norm is in shape of torch.Size([0]), and k_states_image_token'shape is torch.Size([0, 512])