Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-16 16:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix yapf

Browse Source
This commit is contained in:
ly119399
2024-01-10 23:28:19 +08:00
parent 1b59c023bf
commit 6cb4722ffb
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
modelscope/pipelines/nlp/translation_pipeline.py
View File

@@ -51,14 +51,12 @@ class TranslationPipeline(Pipeline):
self._src_vocab_path = osp.join(
model, self.cfg['dataset']['src_vocab']['file'])
self._src_vocab = dict([
(w.strip(), i) for i, w in enumerate(open(self._src_vocab_path, encoding='utf-8'))
])
self._src_vocab = dict([(w.strip(), i) for i, w in enumerate(
open(self._src_vocab_path, encoding='utf-8'))])
self._trg_vocab_path = osp.join(
model, self.cfg['dataset']['trg_vocab']['file'])
self._trg_rvocab = dict([
(i, w.strip()) for i, w in enumerate(open(self._trg_vocab_path, encoding='utf-8'))
])
self._trg_rvocab = dict([(i, w.strip()) for i, w in enumerate(
open(self._trg_vocab_path, encoding='utf-8'))])
tf_config = tf.ConfigProto(allow_soft_placement=True)
tf_config.gpu_options.allow_growth = True