dh-mc commited on
Commit
b6beb8f
·
1 Parent(s): 94283b1

Update eval_shots.py

Browse files
Files changed (1) hide show
  1. llm_toolkit/eval_shots.py +1 -1
llm_toolkit/eval_shots.py CHANGED
@@ -30,7 +30,7 @@ batch_size = int(os.getenv("BATCH_SIZE", 1))
30
  using_llama_factory = os.getenv("USING_LLAMA_FACTORY") == "true"
31
  max_new_tokens = int(os.getenv("MAX_NEW_TOKENS", 2048))
32
  start_num_shots = int(os.getenv("START_NUM_SHOTS", 0))
33
- end_num_shots = int(os.getenv("END_NUM_SHOTS", 50)
34
 
35
  print(
36
  model_name,
 
30
  using_llama_factory = os.getenv("USING_LLAMA_FACTORY") == "true"
31
  max_new_tokens = int(os.getenv("MAX_NEW_TOKENS", 2048))
32
  start_num_shots = int(os.getenv("START_NUM_SHOTS", 0))
33
+ end_num_shots = int(os.getenv("END_NUM_SHOTS", 50))
34
 
35
  print(
36
  model_name,