Final_Assignment_Template

Sleeping

Freddolin commited on Jul 8

Commit

aef8c2b

verified ·

1 Parent(s): 1582f93

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -19,7 +19,7 @@ class GaiaAgent:
             model="google/flan-t5-large",  # <-- Bara denna rad
             token=token,
             device="cpu",
-            max_new_tokens=1024,
             do_sample=False,
             temperature=0.1,
         )

             model="google/flan-t5-large",  # <-- Bara denna rad
             token=token,
             device="cpu",
+            max_new_tokens=256,
             do_sample=False,
             temperature=0.1,
         )