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):
|