Freddolin commited on
Commit
8bfc54d
·
verified ·
1 Parent(s): 21e96d1

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -30,7 +30,7 @@ class GaiaAgent:
30
  "text-generation",
31
  model=self.model,
32
  tokenizer=self.tokenizer,
33
- device=0 if device == "cuda" else -1 # 0 för första GPU, -1 för CPU
34
  )
35
  print("Textgenereringspipeline skapad.")
36
 
 
30
  "text-generation",
31
  model=self.model,
32
  tokenizer=self.tokenizer,
33
+ # device=0 if device == "cuda" else -1 # 0 för första GPU, -1 för CPU
34
  )
35
  print("Textgenereringspipeline skapad.")
36