Update README.md

README.md

@@ -27,7 +27,7 @@ A simple demo for deployment of the model:
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-model_path = "THUDM/LongReward-llama3.1-8b-SFT"
+model_path = "NeoZ123/LongReward-llama3.1-8b-SFT"
 tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=torch.bfloat16, trust_remote_code=True, device_map='auto')
 context = '''
@@ -44,7 +44,7 @@ You can also deploy the model with [vllm](https://github.com/vllm-project/vllm)
 import torch
 from vllm import LLM, SamplingParams
 
-model_path = "THUDM/LongReward-llama3.1-8b-SFT"
+model_path = "NeoZ123/LongReward-llama3.1-8b-SFT"
 model = LLM(
     model= model_path,
     dtype=torch.bfloat16,