Update modules/translation/translation_base.py
Browse files
modules/translation/translation_base.py
CHANGED
@@ -161,8 +161,16 @@ class TranslationBase(ABC):
|
|
161 |
List[dict] with translation
|
162 |
"""
|
163 |
try:
|
164 |
-
self.cache_parameters(model_size=model_size,
|
165 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
166 |
|
167 |
total_progress = len(input_list_dict)
|
168 |
for index, dic in enumerate(input_list_dict):
|
|
|
161 |
List[dict] with translation
|
162 |
"""
|
163 |
try:
|
164 |
+
self.cache_parameters(model_size=model_size,
|
165 |
+
src_lang=src_lang,
|
166 |
+
tgt_lang=tgt_lang,
|
167 |
+
max_length=max_length,
|
168 |
+
add_timestamp=add_timestamp)
|
169 |
+
|
170 |
+
self.update_model(model_size=model_size,
|
171 |
+
src_lang=src_lang,
|
172 |
+
tgt_lang=tgt_lang,
|
173 |
+
progress=progress)
|
174 |
|
175 |
total_progress = len(input_list_dict)
|
176 |
for index, dic in enumerate(input_list_dict):
|