nicholasKluge commited on
Commit
cdff38c
1 Parent(s): 2677dc0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -84,8 +84,9 @@ aira.to(device)
84
 
85
  question = input("Enter your question: ")
86
 
87
- # OPT tokenizer already adds the BOS token, so we do not need to add it manually
88
- inputs = tokenizer(question + tokenizer.sep_token, return_tensors="pt").to(device)
 
89
 
90
  responses = aira.generate(**inputs,
91
  do_sample=True,
 
84
 
85
  question = input("Enter your question: ")
86
 
87
+ inputs = tokenizer(tokenizer.bos_token + question + tokenizer.sep_token,
88
+ add_special_tokens=False,
89
+ return_tensors="pt").to(device)
90
 
91
  responses = aira.generate(**inputs,
92
  do_sample=True,