Commit
·
e26ac7b
1
Parent(s):
2d87ed2
Update README.md
Browse files
README.md
CHANGED
|
@@ -83,12 +83,7 @@ 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,
|
| 88 |
-
top_k=50,
|
| 89 |
-
top_p=0.95,
|
| 90 |
-
temperature=0.7,
|
| 91 |
-
num_return_sequences=2)
|
| 92 |
|
| 93 |
print(f"Question: 👤 {question}\n")
|
| 94 |
|
|
|
|
| 83 |
add_special_tokens=False,
|
| 84 |
return_tensors="pt").to(device)
|
| 85 |
|
| 86 |
+
responses = aira.generate(**inputs, num_return_sequences=2)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 87 |
|
| 88 |
print(f"Question: 👤 {question}\n")
|
| 89 |
|