Update README.md

README.md

@@ -159,7 +159,7 @@ model_name = "MiniMaxAI/MiniMax-Text-01"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, torch_dtype=torch.bfloat16)
 
-fp_layers = [f"model.layers.{i}.block_sparse_moe.gate" for i in range(config.num_hidden_layers)]
+fp_layers = [f"model.layers.{i}.block_sparse_moe.gate" for i in range(model.config.num_hidden_layers)]
 layer_config = {}
 for fp_layer in fp_layers:
     layer_config[fp_layer] = {"bits": 16}