linzhao-amd commited on
Commit
5751a6a
·
verified ·
1 Parent(s): 2c85981

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -126,7 +126,7 @@ lm_eval --model local-completions \
126
  ```
127
  lm_eval \
128
  --model vllm \
129
- --model_args pretrained=amd/DeepSeek-R1-MXFP4,dtype=auto,add_bos_token=True,tensor_parallel_size=$tp_size,gpu_memory_utilization=0.8,max_model_len=38768, \
130
  --tasks gsm8k \
131
  --num_fewshot 8 \
132
  --batch_size auto \
 
126
  ```
127
  lm_eval \
128
  --model vllm \
129
+ --model_args pretrained=amd/DeepSeek-R1-MXFP4,dtype=auto,add_bos_token=True,tensor_parallel_size=8,gpu_memory_utilization=0.8,max_model_len=38768, \
130
  --tasks gsm8k \
131
  --num_fewshot 8 \
132
  --batch_size auto \