Minor fix

Author: mergennachin

examples/models/gemma4_31b/export.py

@@ -162,7 +162,7 @@ def _export_cuda(model: Gemma4_31B, config: Gemma4_31BConfig, output_dir: str) -
 
     # Prefill (T>=2): shim does dequant+cuBLAS (optimal for large M).
     max_prefill = min(config.max_seq_len - 1, config.sliding_window * 2)
-    seq_dim = Dim("seq_len", min=2, max=max_prefill)
+    seq_dim = Dim("seq_len", min=5, max=max_prefill)
     print(f"Exporting prefill (T in [2, {max_prefill}])...")
     with torch.no_grad():
         prefill_ep = export(