davda54 commited on
Commit
a38f207
·
1 Parent(s): cc76a88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -110,16 +110,16 @@ def translate(source, source_language, target_language):
110
  input_ids=source_subwords,
111
  attention_mask=(source_subwords != pad_index).long(),
112
  max_new_tokens = 512-1,
113
- top_k=64,
114
- top_p=0.95,
115
- do_sample=True,
116
- temperature=0.3,
117
  num_beams=1,
118
- use_cache=True,
119
  logits_processor=[RepetitionPenaltyLogitsProcessor(1.0, model)]
120
  # num_beams=4,
121
  # early_stopping=True,
122
- #do_sample=False,
123
  #use_cache=True
124
  )
125
  t = Thread(target=generate, args=(model,), kwargs=generate_kwargs)
 
110
  input_ids=source_subwords,
111
  attention_mask=(source_subwords != pad_index).long(),
112
  max_new_tokens = 512-1,
113
+ #top_k=64,
114
+ #top_p=0.95,
115
+ #do_sample=True,
116
+ #temperature=0.3,
117
  num_beams=1,
118
+ #use_cache=True,
119
  logits_processor=[RepetitionPenaltyLogitsProcessor(1.0, model)]
120
  # num_beams=4,
121
  # early_stopping=True,
122
+ do_sample=False,
123
  #use_cache=True
124
  )
125
  t = Thread(target=generate, args=(model,), kwargs=generate_kwargs)