Updated modeling_auto.py to handle num_kv_blocks=None case gracefully

Signed-off-by: Vaibhav Verma <[email protected]>

Author: vaibverm

QEfficient/transformers/models/modeling_auto.py

@@ -2308,7 +2308,11 @@ def __init__(
         if self.is_tlm:
             self.model.qaic_config["return_pdfs"] = True
 
-        num_kv_blocks = self.model.qaic_config.get("num_kv_blocks", None)
+        if self.model.qaic_config is not None:
+            num_kv_blocks = self.model.qaic_config.get("num_kv_blocks", None)
+        else:
+            num_kv_blocks = None
+
         if num_kv_blocks is not None:
             BlockedKVAttentionTransform.apply(model, num_kv_blocks=num_kv_blocks)