Freddolin commited on
Commit
b7ae39c
·
verified ·
1 Parent(s): 8b68d06

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -17,7 +17,7 @@ class GaiaAgent:
17
  self.llm = pipeline(
18
  "text-generation",
19
  model="microsoft/DialoGPT-medium", # <-- Bara denna rad
20
- use_auth_token=token,
21
  device="cpu",
22
  max_new_tokens=1024,
23
  do_sample=False,
 
17
  self.llm = pipeline(
18
  "text-generation",
19
  model="microsoft/DialoGPT-medium", # <-- Bara denna rad
20
+ token=token,
21
  device="cpu",
22
  max_new_tokens=1024,
23
  do_sample=False,