mgoin commited on
Commit
1c04e39
1 Parent(s): f0ca39a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -0
README.md CHANGED
@@ -10,6 +10,12 @@ vllm (pretrained=/home/mgoin/code/Nemotron-4-340B-Base-hf-FP8,tensor_parallel_si
10
  | Tasks |Version|Filter|n-shot|Metric| |Value | |Stderr|
11
  |--------------|------:|------|-----:|------|---|-----:|---|-----:|
12
  |truthfulqa_mc2| 2|none | 0|acc |↑ |0.4869|± |0.0142|
 
 
 
 
 
 
13
  ```
14
 
15
  The [original paper](https://arxiv.org/pdf/2407.14679) evals:
 
10
  | Tasks |Version|Filter|n-shot|Metric| |Value | |Stderr|
11
  |--------------|------:|------|-----:|------|---|-----:|---|-----:|
12
  |truthfulqa_mc2| 2|none | 0|acc |↑ |0.4869|± |0.0142|
13
+
14
+ lm_eval --model vllm --model_args pretrained=/home/mgoin/code/Nemotron-4-340B-Base-hf-FP8,tensor_parallel_size=8,distributed_executor_backend="ray",max_model_len=4096,gpu_memory_utilization=0.6 --tasks winogrande --num_fewshot 5 --batch_size 16
15
+ vllm (pretrained=/home/mgoin/code/Nemotron-4-340B-Base-hf-FP8,tensor_parallel_size=8,distributed_executor_backend=ray,max_model_len=4096,gpu_memory_utilization=0.6), gen_kwargs: (None), limit: None, num_fewshot: 5, batch_size: 16
16
+ | Tasks |Version|Filter|n-shot|Metric| |Value | |Stderr|
17
+ |----------|------:|------|-----:|------|---|-----:|---|-----:|
18
+ |winogrande| 1|none | 5|acc |↑ |0.8887|± |0.0088|
19
  ```
20
 
21
  The [original paper](https://arxiv.org/pdf/2407.14679) evals: