Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
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)
|