nicholasKluge
commited on
Commit
•
cdff38c
1
Parent(s):
2677dc0
Update README.md
Browse files
README.md
CHANGED
@@ -84,8 +84,9 @@ aira.to(device)
|
|
84 |
|
85 |
question = input("Enter your question: ")
|
86 |
|
87 |
-
|
88 |
-
|
|
|
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,
|