jhj0517 commited on
Commit
d8c2d87
·
1 Parent(s): bf46b63
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
- "src_lang": src_lang,
171
- "tgt_lang": tgt_lang,
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