Update README.md

README.md

@@ -70,8 +70,8 @@ We can perform inference for the Skywork-MoE-base (16x13B size) model using Hugg
 
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model = AutoModelForCausalLM.from_pretrained("Skywork/Skywork-MoE-base", trust_remote_code=True, device_map='auto')
-tokenizer = AutoTokenizer.from_pretrained("Skywork/Skywork-MoE-base", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Skywork/Skywork-MoE-Base", trust_remote_code=True, device_map='auto')
+tokenizer = AutoTokenizer.from_pretrained("Skywork/Skywork-MoE-Base", trust_remote_code=True)
 
 inputs = tokenizer('陕西的省会是西安', return_tensors='pt').to(model.device)
 response = model.generate(inputs.input_ids, max_length=128)