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,
|