drrobot9 commited on
Commit
6d446ee
·
verified ·
1 Parent(s): e600841

Update app/agents/crew_pipeline.py

Browse files
Files changed (1) hide show
  1. app/agents/crew_pipeline.py +1 -1
app/agents/crew_pipeline.py CHANGED
@@ -226,7 +226,7 @@ def run_qwen(messages: List[dict], max_new_tokens: int = 1300) -> str:
226
  generated_ids = model.generate(
227
  **inputs,
228
  max_new_tokens=max_new_tokens,
229
- temperature=0.4,
230
  repetition_penalty=1.1
231
  )
232
  output_ids = generated_ids[0][len(inputs.input_ids[0]):].tolist()
 
226
  generated_ids = model.generate(
227
  **inputs,
228
  max_new_tokens=max_new_tokens,
229
+ temperature=0.2,
230
  repetition_penalty=1.1
231
  )
232
  output_ids = generated_ids[0][len(inputs.input_ids[0]):].tolist()