nicholasKluge commited on
Commit
39fedf1
1 Parent(s): ac9fca3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -79,7 +79,9 @@ aira.to(device)
79
 
80
  question = input("Enter your question: ")
81
 
82
- inputs = tokenizer(tokenizer.bos_token + question + tokenizer.sep_token, return_tensors="pt").to(device)
 
 
83
 
84
  responses = aira.generate(**inputs,
85
  do_sample=True,
 
79
 
80
  question = input("Enter your question: ")
81
 
82
+ inputs = tokenizer(tokenizer.bos_token + question + tokenizer.sep_token,
83
+ add_special_tokens=False,
84
+ return_tensors="pt").to(device)
85
 
86
  responses = aira.generate(**inputs,
87
  do_sample=True,