Manel commited on
Commit
24e6bd1
·
verified ·
1 Parent(s): 8c0109f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ def wrap_model(model, tokenizer):
88
  task="text-generation",
89
  temperature=0.5,
90
  repetition_penalty=2.1,
91
- no_repeat_ngram_size=3
92
  max_new_tokens=400,
93
  num_beams=2,
94
  pad_token_id=2,
 
88
  task="text-generation",
89
  temperature=0.5,
90
  repetition_penalty=2.1,
91
+ no_repeat_ngram_size=3,
92
  max_new_tokens=400,
93
  num_beams=2,
94
  pad_token_id=2,