osanseviero HF staff commited on
Commit
23e43c0
1 Parent(s): b18f6c8

Fix char.dict finding

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -19,7 +19,7 @@ class PreTrainedModel(Runner):
19
  ckp["Args"].init_ckpt = ckp_file
20
  ckp["Args"].mode = "inference"
21
  ckp["Args"].device = "cpu" # Just to try in my computer
22
- ckp["Config"]["downstream_expert"]["datarc"]["dict_path"]='char.dict'
23
 
24
  Runner.__init__(self, ckp["Args"], ckp["Config"])
25
 
19
  ckp["Args"].init_ckpt = ckp_file
20
  ckp["Args"].mode = "inference"
21
  ckp["Args"].device = "cpu" # Just to try in my computer
22
+ ckp["Config"]["downstream_expert"]["datarc"]["dict_path"]=os.path.join(path,'char.dict')
23
 
24
  Runner.__init__(self, ckp["Args"], ckp["Config"])
25