Update agent.py
Browse files
agent.py
CHANGED
@@ -369,12 +369,12 @@ from llama_index.llms.huggingface import HuggingFaceLLM
|
|
369 |
# This model is loaded only once for efficiency.
|
370 |
try:
|
371 |
code_llm = HuggingFaceLLM(
|
372 |
-
model_name="Qwen/Qwen2.5-Coder-
|
373 |
-
tokenizer_name="Qwen/Qwen2.5-Coder-
|
374 |
device_map="auto",
|
375 |
model_kwargs={"torch_dtype": "auto"},
|
376 |
# Set generation parameters for precise, non-creative code output
|
377 |
-
generate_kwargs={"temperature": 0.
|
378 |
)
|
379 |
except Exception as e:
|
380 |
print(f"Error initializing code generation model: {e}")
|
|
|
369 |
# This model is loaded only once for efficiency.
|
370 |
try:
|
371 |
code_llm = HuggingFaceLLM(
|
372 |
+
model_name="Qwen/Qwen2.5-Coder-3B",
|
373 |
+
tokenizer_name="Qwen/Qwen2.5-Coder-3B",
|
374 |
device_map="auto",
|
375 |
model_kwargs={"torch_dtype": "auto"},
|
376 |
# Set generation parameters for precise, non-creative code output
|
377 |
+
generate_kwargs={"temperature": 0.1, "do_sample": False}
|
378 |
)
|
379 |
except Exception as e:
|
380 |
print(f"Error initializing code generation model: {e}")
|