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
|