Update README.md
Browse files
README.md
CHANGED
@@ -52,7 +52,7 @@ def to_markdown(text):
|
|
52 |
# Generate text
|
53 |
output = model.generate(
|
54 |
input_ids,
|
55 |
-
max_length=
|
56 |
num_return_sequences=1, # Number of generated responses
|
57 |
no_repeat_ngram_size=2, # Prevent repetition
|
58 |
top_k=50, # Filter to top-k tokens
|
|
|
52 |
# Generate text
|
53 |
output = model.generate(
|
54 |
input_ids,
|
55 |
+
max_length=128, # Adjust max length as needed
|
56 |
num_return_sequences=1, # Number of generated responses
|
57 |
no_repeat_ngram_size=2, # Prevent repetition
|
58 |
top_k=50, # Filter to top-k tokens
|