fix smol mistake
Browse files
README.md
CHANGED
@@ -68,7 +68,7 @@ input_document = "Alice was beginning to get very tired of sitting by her sister
|
|
68 |
|
69 |
inputs = tokenizer(prefix + input_document, return_tensors="pt", max_length=16384, truncation=True, add_special_tokens=True)
|
70 |
|
71 |
-
outputs = model.generate(inputs
|
72 |
|
73 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
74 |
```
|
|
|
68 |
|
69 |
inputs = tokenizer(prefix + input_document, return_tensors="pt", max_length=16384, truncation=True, add_special_tokens=True)
|
70 |
|
71 |
+
outputs = model.generate(**inputs, **gen_kwargs)
|
72 |
|
73 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
74 |
```
|