Upload folder using huggingface_hub

README.md

@@ -126,17 +126,14 @@ model.model.ngram_embeddings = None  # avoid saving shared params
 model.save_pretrained(save_folder)
 torch.set_default_dtype(torch.float32)
 
-print(model.model.rotary_emb.inv_freq.shape)
-# 1 / 0
-# for n, m in model.named_modules():
-#     if 'LongcatFlashMLA' in str(type(m)):
-#         print(n, m.layer_idx)
-
 with open(f"{save_folder}/config.json", "r", encoding='utf-8') as f:
     config_json = json.load(f)
-    config_json['auto_map'] = {k: v.split('--')[-1] for k, v in config_json['auto_map'].items()}
+    config_json['auto_map'] = {k: source_model_id + '--' +
+                               v.split('--')[-1] for k, v in config_json['auto_map'].items()}
 with open(f"{save_folder}/config.json", "w", encoding='utf-8') as f:
     json.dump(config_json, f, indent=2)
+for f in Path(save_folder).glob('*.py'):
+    f.unlink()
 ```
 
 </details>

config.json

@@ -5,9 +5,9 @@
   "attention_bias": false,
   "attention_dropout": 0.0,
   "auto_map": {
-    "AutoConfig": "configuration_longcat_ngram.LongcatFlashNgramConfig",
-    "AutoModel": "modeling_longcat_ngram.LongcatFlashNgramModel",
-    "AutoModelForCausalLM": "modeling_longcat_ngram.LongcatFlashNgramForCausalLM"
+    "AutoConfig": "meituan-longcat/LongCat-Flash-Lite--configuration_longcat_ngram.LongcatFlashNgramConfig",
+    "AutoModel": "meituan-longcat/LongCat-Flash-Lite--modeling_longcat_ngram.LongcatFlashNgramModel",
+    "AutoModelForCausalLM": "meituan-longcat/LongCat-Flash-Lite--modeling_longcat_ngram.LongcatFlashNgramForCausalLM"
   },
   "bos_token_id": 1,
   "dtype": "bfloat16",