Fix code formats

Author: yzhangcs

configs/nsa_340M.json

@@ -1,13 +1,5 @@
 {
     "attention_bias": false,
-    "architectures": [
-        "NSAForCausalLM"
-    ],
-    "auto_map": {
-        "AutoConfig": "native_sparse_attention.NSAConfig",
-        "AutoModel": "native_sparse_attention.NSAModel",
-        "AutoModelForCausalLM": "native_sparse_attention.NSAForCausalLM"
-    },
     "bos_token_id": 1,
     "eos_token_id": 2,
     "fuse_cross_entropy": true,
@@ -27,4 +19,5 @@
     "tie_word_embeddings": false,
     "use_cache": true,
     "vocab_size": 32000
-}
+}
+