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

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -16,10 +16,10 @@ class GaiaAgent:
16
  # Använd en mer kapabel modell för bättre reasoning
17
  self.llm = pipeline(
18
  "text-generation",
19
- model="mistralai/Mistral-7B-Instruct-v0.2",
20
  use_auth_token=token,
21
  device="cpu",
22
- max_new_tokens=1024, # Öka för mer detaljerade svar
23
  do_sample=False,
24
  temperature=0.1,
25
  return_full_text=False
 
16
  # Använd en mer kapabel modell för bättre reasoning
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,
24
  temperature=0.1,
25
  return_full_text=False