SaviAnna commited on
Commit
073629e
·
verified ·
1 Parent(s): 1c8cc34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def generate_text(model, tokenizer, prompt, max_len, temperature):
38
  repetition_penalty=1.2, # Штраф за повторение слов или фраз
39
  no_repeat_ngram_size=4, # Запрет на повторение n-грамм (например, биграмм)
40
  do_sample=True, # Включение сэмплинга для большей разнообразности
41
- attention_mask=attention_mask
42
  pad_token_id=tokenizer.eos_token_id
43
  )
44
 
 
38
  repetition_penalty=1.2, # Штраф за повторение слов или фраз
39
  no_repeat_ngram_size=4, # Запрет на повторение n-грамм (например, биграмм)
40
  do_sample=True, # Включение сэмплинга для большей разнообразности
41
+ attention_mask=attention_mask,
42
  pad_token_id=tokenizer.eos_token_id
43
  )
44