ai_agent_course_final_project

Running

keynes42 commited on May 20

Commit

a250a24

verified ·

1 Parent(s): 0db4386

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -98,6 +98,7 @@ class BasicAgent:
             model=mod,
             tokenizer=tok,
             max_new_tokens=512,
             # temperature=1.0,
         )
         self.agent = CodeAgent(model=self, tools=[], add_base_tools=True)

             model=mod,
             tokenizer=tok,
             max_new_tokens=512,
+            return_full_text=False,      # <— only get the completion, not the prompt + completion
             # temperature=1.0,
         )
         self.agent = CodeAgent(model=self, tools=[], add_base_tools=True)