JunhanCai commited on
Commit
fdba9e5
·
verified ·
1 Parent(s): 8f6b390

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. webserver/train_service.py +1 -1
webserver/train_service.py CHANGED
@@ -323,7 +323,7 @@ def run_finetune_job(job_id: str, input_paths: dict, run_dir: str, config: Train
323
  )
324
 
325
  classifier, _, _, mae_model = load_mae_model_for_classification(
326
- input_paths["model"],
327
  input_length=3500,
328
  patch_num=config.patch_num,
329
  embedding_dim=config.embedding_dim,
 
323
  )
324
 
325
  classifier, _, _, mae_model = load_mae_model_for_classification(
326
+ model_path,
327
  input_length=3500,
328
  patch_num=config.patch_num,
329
  embedding_dim=config.embedding_dim,