Specify model name

model.py

@@ -19,6 +19,7 @@ class PreTrainedModel(Runner):
         ckp["Args"].init_ckpt = ckp_file
         ckp["Args"].mode = "inference"
         ckp["Args"].upstream = "osanseviero/hubert_base"
+        ckp["Args"].upstream_model_name = "model.pt"
         ckp["Args"].from_hf_hub = True
         ckp["Args"].device = "cpu" # Just to try in my computer
         ckp["Config"]["downstream_expert"]["datarc"]["dict_path"]=os.path.join(path,'char.dict')