config.json

{
  "base_model_name_or_path": "/root/paddlejob/workspace/output/attn_medusa/pt_model",
  "conv_kernel_size": 1,
  "extract_layers": [
    -1
  ],
  "medusa_num_heads": 4,
  "medusa_num_layers": 1,
  "small_model_name_or_path": "/root/paddlejob/workspace/output/MAR/tiny_model",
  "transformers_version": "4.57.0"
}