osanseviero
HF staff
commited on
Commit
3020f73
1 Parent(s): 7cfa43d

Specify model name

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