Commit
•
995c76a
1
Parent(s):
af3556d
Update README.md
Browse files
README.md
CHANGED
|
@@ -82,12 +82,8 @@ question = input("Enter your question: ")
|
|
| 82 |
inputs = tokenizer(tokenizer.bos_token + question + tokenizer.sep_token, return_tensors="pt").to(device)
|
| 83 |
|
| 84 |
responses = aira.generate(**inputs,
|
| 85 |
-
bos_token_id=tokenizer.bos_token_id,
|
| 86 |
-
pad_token_id=tokenizer.pad_token_id,
|
| 87 |
-
eos_token_id=tokenizer.eos_token_id,
|
| 88 |
do_sample=True,
|
| 89 |
top_k=50,
|
| 90 |
-
max_length=500,
|
| 91 |
top_p=0.95,
|
| 92 |
temperature=0.7,
|
| 93 |
num_return_sequences=2)
|
|
|
|
| 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,
|
| 86 |
top_k=50,
|
|
|
|
| 87 |
top_p=0.95,
|
| 88 |
temperature=0.7,
|
| 89 |
num_return_sequences=2)
|