Revert "remove change breaking sharktank benchmarks"

This reverts commit c617cfb.

sharktank/sharktank/examples/export_paged_llm_v1.py

@@ -96,9 +96,9 @@ def generate_params_json(hp, prefill_bs: list[int], decode_bs: list[int]):
             "prefill_batch_sizes": prefill_bs,
             "decode_batch_sizes": decode_bs,
             "transformer_block_count": hp.block_count,
-            "block_seq_stride": llama_config.block_seq_stride,
             "paged_kv_cache": {
                 "attention_head_count_kv": hp.attention_head_count_kv,
+                "block_seq_stride": llama_config.block_seq_stride,
                 "device_block_count": 256,  # so that this makes its way into the config file & can be edited.
             },
         }