77ethers commited on
Commit
aee53e2
·
verified ·
1 Parent(s): bfb3f8a

Upload inference.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 = 150
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.