Commit
·
4e55943
1
Parent(s):
d10a712
Update README.md
Browse files
README.md
CHANGED
@@ -88,9 +88,6 @@ question = input("Enter your question: ")
|
|
88 |
inputs = tokenizer(tokenizer.bos_token + question + tokenizer.sep_token, return_tensors="pt").to(device)
|
89 |
|
90 |
responses = aira.generate(**inputs,
|
91 |
-
bos_token_id=tokenizer.bos_token_id,
|
92 |
-
pad_token_id=tokenizer.pad_token_id,
|
93 |
-
eos_token_id=tokenizer.eos_token_id,
|
94 |
do_sample=True,
|
95 |
top_k=50,
|
96 |
max_length=500,
|
|
|
88 |
inputs = tokenizer(tokenizer.bos_token + question + tokenizer.sep_token, return_tensors="pt").to(device)
|
89 |
|
90 |
responses = aira.generate(**inputs,
|
|
|
|
|
|
|
91 |
do_sample=True,
|
92 |
top_k=50,
|
93 |
max_length=500,
|