edwko commited on
Commit
d478a4a
1 Parent(s): 39379dc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -238,7 +238,7 @@ def generate_response(message: str, temperature: float = 0.1, repetition_penalty
238
  # Generate the response
239
  output = model.generate(
240
  input_ids,
241
- max_length=512,
242
  temperature=temperature,
243
  repetition_penalty=repetition_penalty,
244
  do_sample=True
 
238
  # Generate the response
239
  output = model.generate(
240
  input_ids,
241
+ max_length=256,
242
  temperature=temperature,
243
  repetition_penalty=repetition_penalty,
244
  do_sample=True