Update configuration_decilm.py

configuration_decilm.py

@@ -32,7 +32,7 @@ class DeciLMConfig(LlamaConfig):
 
     def __init__(
         self,
-        num_key_value_heads_per_layer: list[int] = None,
+        num_key_value_heads_per_layer: list = None,
         naive_attention_prefill: bool = False,
         naive_attention_decode_batched: bool = False,
         naive_attention_decode_single: bool = False,