Update README.md

README.md

@@ -39,7 +39,7 @@ from transformers import AutoTokenizer
 from vllm import LLM, SamplingParams
 
 max_model_len, tp_size = 4096, 1
-model_name = "neuralmagic-ent/DeepSeek-R1-Distill-Llama-8B-FP8-Dynamic"
+model_name = "neuralmagic/DeepSeek-R1-Distill-Llama-8B-FP8-Dynamic"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 llm = LLM(model=model_name, tensor_parallel_size=tp_size, max_model_len=max_model_len, trust_remote_code=True)
 sampling_params = SamplingParams(temperature=0.6, max_tokens=256, stop_token_ids=[tokenizer.eos_token_id])
@@ -112,7 +112,7 @@ OpenLLM Leaderboard V1:
 ```
 lm_eval \
   --model vllm \
-  --model_args pretrained="neuralmagic-ent/DeepSeek-R1-Distill-Llama-8B-FP8-Dynamic",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=1,gpu_memory_utilization=0.8,enable_chunked_prefill=True,trust_remote_code=True \
+  --model_args pretrained="neuralmagic/DeepSeek-R1-Distill-Llama-8B-FP8-Dynamic",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=1,gpu_memory_utilization=0.8,enable_chunked_prefill=True,trust_remote_code=True \
   --tasks openllm \
   --write_out \
   --batch_size auto \
@@ -124,7 +124,7 @@ OpenLLM Leaderboard V2:
 ```
 lm_eval \
   --model vllm \
-  --model_args pretrained="neuralmagic-ent/DeepSeek-R1-Distill-Llama-8B-FP8-Dynamic",dtype=auto,add_bos_token=False,max_model_len=4096,tensor_parallel_size=1,gpu_memory_utilization=0.8,enable_chunked_prefill=True,trust_remote_code=True \
+  --model_args pretrained="neuralmagic/DeepSeek-R1-Distill-Llama-8B-FP8-Dynamic",dtype=auto,add_bos_token=False,max_model_len=4096,tensor_parallel_size=1,gpu_memory_utilization=0.8,enable_chunked_prefill=True,trust_remote_code=True \
   --apply_chat_template \
   --fewshot_as_multiturn \
   --tasks leaderboard \
@@ -143,7 +143,7 @@ lm_eval \
       <th>Category</th>
       <th>Metric</th>
       <th>deepseek-ai/DeepSeek-R1-Distill-Llama-8B</th>
-      <th>neuralmagic-ent/DeepSeek-R1-Distill-Llama-8B-FP8-Dynamic</th>
+      <th>neuralmagic/DeepSeek-R1-Distill-Llama-8B-FP8-Dynamic</th>
       <th>Recovery</th>
     </tr>
   </thead>