shivansarora commited on
Commit
2856fde
·
verified ·
1 Parent(s): 3643648

minimal beam test

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -71,6 +71,8 @@ def chat(user_input, cefr, inference_type, history):
71
 
72
  print("🔥 Warming up...")
73
  conversation_state["human_agent"].msg = "Hello"
 
 
74
  conversation_state["world"].parley()
75
  print("✅ Warmup complete.")
76
 
 
71
 
72
  print("🔥 Warming up...")
73
  conversation_state["human_agent"].msg = "Hello"
74
+ conversation_state["world"].agents[1].opt['beam_size'] = 1
75
+ conversation_state["world"].agents[1].opt['topk'] = 10
76
  conversation_state["world"].parley()
77
  print("✅ Warmup complete.")
78