Update README.md
#3
by
Tonic
- opened
README.md
CHANGED
@@ -44,7 +44,7 @@ model.eval()
|
|
44 |
|
45 |
prompt = 'Q: What is the largest bird?\nA:'
|
46 |
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
|
47 |
-
tokens = model.generate(input_ids, max_length=
|
48 |
print( tokenizer.decode(tokens[0].tolist(), skip_special_tokens=True) )
|
49 |
# Q: What is the largest bird?\nA: The largest bird is a black-headed gull.
|
50 |
```
|
|
|
44 |
|
45 |
prompt = 'Q: What is the largest bird?\nA:'
|
46 |
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
|
47 |
+
tokens = model.generate(input_ids, max_length=22)
|
48 |
print( tokenizer.decode(tokens[0].tolist(), skip_special_tokens=True) )
|
49 |
# Q: What is the largest bird?\nA: The largest bird is a black-headed gull.
|
50 |
```
|