wcosmas commited on
Commit
70b08cd
·
verified ·
1 Parent(s): 9de05fb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -119,7 +119,7 @@ def generate_sbcc_message(practice, message_type="adoption", barrier=None, defic
119
  with torch.no_grad():
120
  outputs = model.generate(
121
  **inputs,
122
- max_new_tokens=130,
123
  temperature=0.3,
124
  do_sample=True,
125
  pad_token_id=tokenizer.eos_token_id
 
119
  with torch.no_grad():
120
  outputs = model.generate(
121
  **inputs,
122
+ max_new_tokens=300,
123
  temperature=0.3,
124
  do_sample=True,
125
  pad_token_id=tokenizer.eos_token_id