Spaces:
Runtime error
Runtime error
Upload folder using huggingface_hub
Browse files
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 |
-
|
| 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,
|