added opt swa to hash

Signed-off-by: Onkar Chougule <ochougul@qti.qualcomm.com>

Author: ochougul

QEfficient/transformers/models/modeling_auto.py

@@ -2343,8 +2343,9 @@ def get_seq_len_and_handle_specialized_prefill_model(self, prefill_seq_len: Opti
 
         self.prefill(True)
         self.hash_params["prefill_only"] = True
-        self.hash_params["num_blocks"] = num_q_blocks
-        self.hash_params["num_ffn_blocks"] = num_ffn_blocks
+        self.hash_params["NUM_Q_BLOCKS"] = num_q_blocks
+        self.hash_params["NUM_FFN_BLOCKS"] = num_ffn_blocks
+        self.hash_params["ENABLE_OPT_SWA"] = os.environ.get("ENABLE_OPT_SWA", "0")
         return (
             min_seq_len
             if min_seq_len > constants.ONNX_EXPORT_EXAMPLE_SEQ_LEN