Skip to content

Commit 0fea3c5

Browse files
authored
Update throughput_config.yaml
1 parent f09c681 commit 0fea3c5

1 file changed

Lines changed: 16 additions & 2 deletions

File tree

leaderboard/throughput_config.yaml

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,21 @@
11
defaults:
2-
- base_config # inherits from base config
2+
- backend: pytorch # default backend
3+
- benchmark: inference # default benchmark
4+
- experiment # inheriting experiment schema
35
- _self_ # for hydra 1.1 compatibility
6+
- override hydra/job_logging: colorlog # colorful logging
7+
- override hydra/hydra_logging: colorlog # colorful logging
48

9+
hydra:
10+
run:
11+
dir: runs/${experiment_name}
12+
sweep:
13+
dir: sweeps/${experiment_name}
14+
job:
15+
chdir: true
16+
env_set:
17+
CUDA_VISIBLE_DEVICES: 0,1,2,3,4,5,6,7
18+
519
experiment_name: code_evals
620

721
model: bigcode/santacoder
@@ -20,4 +34,4 @@ benchmark:
2034
input_shapes:
2135
batch_size: 1
2236
sequence_length: 1
23-
new_tokens: 1000
37+
new_tokens: 1000

0 commit comments

Comments
 (0)