RalphThings commited on
Commit
5bfbd71
·
verified ·
1 Parent(s): 584b7c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -21,8 +21,10 @@ class BasicAgent:
21
  self.generator = pipeline(
22
  "text-generation",
23
  model="gpt2",
24
- framework="pt",
25
- use_auth_token=HF_TOKEN,
 
 
26
  )
27
  # The GAIA system prompt (no "FINAL ANSWER:" at the end)
28
  self.system_prompt = (
 
21
  self.generator = pipeline(
22
  "text-generation",
23
  model="gpt2",
24
+ tokenizer="gpt2",
25
+ framework="pt", # force PyTorch
26
+ token=HF_TOKEN,
27
+ device=-1
28
  )
29
  # The GAIA system prompt (no "FINAL ANSWER:" at the end)
30
  self.system_prompt = (