AksharaSachin commited on
Commit
7d1187c
·
verified ·
1 Parent(s): c9bb62b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -47,7 +47,7 @@ def build_agent():
47
  # huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
48
  # )
49
 
50
- llm = HuggingFaceHub(repo_id="google/flan-t5-xl", task="text-generation",
51
  model_kwargs={
52
  "temperature": 0.7, # Controls randomness (0.0 = deterministic)
53
  "max_new_tokens": 100 # Optional: max output length
 
47
  # huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
48
  # )
49
 
50
+ llm = HuggingFaceHub(repo_id="Qwen/Qwen2.5-Coder-32B-Instruct", task="text-generation",
51
  model_kwargs={
52
  "temperature": 0.7, # Controls randomness (0.0 = deterministic)
53
  "max_new_tokens": 100 # Optional: max output length