Update app/agents/crew_pipeline.py
Browse files
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.
|
| 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()
|