Coool2 commited on
Commit
c6a9f91
·
verified ·
1 Parent(s): 51133ed

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
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-7B",
373
- tokenizer_name="Qwen/Qwen2.5-Coder-7B",
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.0, "do_sample": False}
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}")