Update agent.py
Browse files
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 |
-
|
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,
|