LAP-DEV commited on
Commit
a6bb691
·
verified ·
1 Parent(s): cc50d18

Update modules/translation/translation_base.py

Browse files
modules/translation/translation_base.py CHANGED
@@ -137,7 +137,7 @@ class TranslationBase(ABC):
137
  self.release_cuda_memory()
138
 
139
  def translate_text(self,
140
- List: input_list_dict,
141
  model_size: str,
142
  src_lang: str,
143
  tgt_lang: str,
 
137
  self.release_cuda_memory()
138
 
139
  def translate_text(self,
140
+ input_list_dict: list,
141
  model_size: str,
142
  src_lang: str,
143
  tgt_lang: str,