Commit
•
3020f73
1
Parent(s):
7cfa43d
Specify model name
Browse files
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')
|