I wonder if you could prune the kv cache as well