Final_Assignment_Template

Sleeping

EtienneB commited on Jul 1

Commit

61cae63

1 Parent(s): 042d1d5

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -97,10 +97,9 @@ def build_graph():
     # First create the HuggingFaceEndpoint
     llm_endpoint = HuggingFaceEndpoint(
-        repo_id="gemini/gemini-2.0-flash",
-        #"Qwen/Qwen2.5-Coder-32B-Instruct",
-        #huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
-        api_key=GEMINI_API_KEY,
         temperature=0.3,
         max_new_tokens=512,
         timeout=60,

     # First create the HuggingFaceEndpoint
     llm_endpoint = HuggingFaceEndpoint(
+        repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
+        #api_key=GEMINI_API_KEY,
         temperature=0.3,
         max_new_tokens=512,
         timeout=60,