Spaces:
Build error
Build error
Update eval_shots.py
Browse files
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,
|