gorkemozkaya commited on
Commit
8e7bfe1
1 Parent(s): e542316

fix more path issues

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -8,13 +8,13 @@ from official.core import exp_factory, task_factory
8
  from official.nlp.configs import wmt_transformer_experiments as wmt_te
9
 
10
  tokenizer= tftxt.SentencepieceTokenizer(
11
- model=tf.io.gfile.GFile("/home/user/app/pretrained_v2/sentencepiece_en_tr.model", "rb").read(),
12
  add_eos=True)
13
 
14
  # sys.path = ['/root/.local/lib/python3.10/site-packages', '/root/.local/bin'] + sys.path
15
 
16
  task_config = exp_factory.get_exp_config('transformer_tr_en_blended/base').task
17
- task_config.sentencepiece_model_path = 'pretrained_v2/sentencepiece_en_tr.model'
18
 
19
  translation_task = task_factory.get_task(task_config)
20
  model_en_tr = translation_task.build_model()
 
8
  from official.nlp.configs import wmt_transformer_experiments as wmt_te
9
 
10
  tokenizer= tftxt.SentencepieceTokenizer(
11
+ model=tf.io.gfile.GFile("/code/pretrained_v2/sentencepiece_en_tr.model", "rb").read(),
12
  add_eos=True)
13
 
14
  # sys.path = ['/root/.local/lib/python3.10/site-packages', '/root/.local/bin'] + sys.path
15
 
16
  task_config = exp_factory.get_exp_config('transformer_tr_en_blended/base').task
17
+ task_config.sentencepiece_model_path = '/code/pretrained_v2/sentencepiece_en_tr.model'
18
 
19
  translation_task = task_factory.get_task(task_config)
20
  model_en_tr = translation_task.build_model()