Spaces:
Running
Running
jhj0517
commited on
Commit
·
d8c2d87
1
Parent(s):
bf46b63
Fix key
Browse files
modules/translation/translation_base.py
CHANGED
|
@@ -167,8 +167,8 @@ class TranslationBase(ABC):
|
|
| 167 |
cached_params = load_yaml(DEFAULT_PARAMETERS_CONFIG_PATH)
|
| 168 |
cached_params["translation"]["nllb"] = {
|
| 169 |
"model_size": model_size,
|
| 170 |
-
"
|
| 171 |
-
"
|
| 172 |
"max_length": max_length,
|
| 173 |
}
|
| 174 |
cached_params["translation"]["add_timestamp"] = add_timestamp
|
|
|
|
| 167 |
cached_params = load_yaml(DEFAULT_PARAMETERS_CONFIG_PATH)
|
| 168 |
cached_params["translation"]["nllb"] = {
|
| 169 |
"model_size": model_size,
|
| 170 |
+
"source_lang": src_lang,
|
| 171 |
+
"target_lang": tgt_lang,
|
| 172 |
"max_length": max_length,
|
| 173 |
}
|
| 174 |
cached_params["translation"]["add_timestamp"] = add_timestamp
|