IkIzma commited on
Commit
39f9d41
·
1 Parent(s): 8ea2da0

Update Generate_text.py

Browse files
Files changed (1) hide show
  1. Generate_text.py +1 -1
Generate_text.py CHANGED
@@ -11,7 +11,7 @@ def load_tokenizer_and_model(model_name_or_path, device):
11
 
12
  def generate(
13
  model, tok, text, device,
14
- do_sample=True, max_length=50, repetition_penalty=5.0,
15
  top_k=5, top_p=0.95, temperature=1,
16
  num_beams=None,
17
  no_repeat_ngram_size=3
 
11
 
12
  def generate(
13
  model, tok, text, device,
14
+ do_sample=True, max_length=200, repetition_penalty=5.0,
15
  top_k=5, top_p=0.95, temperature=1,
16
  num_beams=None,
17
  no_repeat_ngram_size=3