Upload inference.py with huggingface_hub
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -30,7 +30,7 @@ from gridops.models import GridOpsAction
|
|
| 30 |
TASKS = ["task_1_normal", "task_2_heatwave", "task_3_crisis"]
|
| 31 |
MAX_STEPS = 72
|
| 32 |
TEMPERATURE = 0.1
|
| 33 |
-
MAX_TOKENS =
|
| 34 |
|
| 35 |
SYSTEM_PROMPT = """\
|
| 36 |
You are an expert microgrid operator managing a 100-home community in India during summer.
|
|
|
|
| 30 |
TASKS = ["task_1_normal", "task_2_heatwave", "task_3_crisis"]
|
| 31 |
MAX_STEPS = 72
|
| 32 |
TEMPERATURE = 0.1
|
| 33 |
+
MAX_TOKENS = 500 # higher to support reasoning models that emit thinking before JSON
|
| 34 |
|
| 35 |
SYSTEM_PROMPT = """\
|
| 36 |
You are an expert microgrid operator managing a 100-home community in India during summer.
|