Update README.md
Browse files
README.md
CHANGED
|
@@ -106,7 +106,7 @@ text = """
|
|
| 106 |
|
| 107 |
# Tokenize input and generate output
|
| 108 |
inputs = tokenizer(text, return_tensors="pt")
|
| 109 |
-
outputs = model.generate(**inputs.to(device), max_new_tokens=
|
| 110 |
|
| 111 |
# Decode the output
|
| 112 |
result = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
|
|
|
| 106 |
|
| 107 |
# Tokenize input and generate output
|
| 108 |
inputs = tokenizer(text, return_tensors="pt")
|
| 109 |
+
outputs = model.generate(**inputs.to(device), max_new_tokens=185)
|
| 110 |
|
| 111 |
# Decode the output
|
| 112 |
result = tokenizer.decode(outputs[0], skip_special_tokens=True)
|