Spaces:

Vishwas1
/

LLMTrainingPro

Sleeping

Vishwas1 commited on Sep 17, 2024

Commit

1a39537

verified ·

1 Parent(s): 21a5890

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -71,7 +71,7 @@ def train_model(request: TrainingRequest):
             "python", TRAIN_MODEL_PATH,
             "--task", request.task,
             "--model_name", request.model_name,
-            "--dataset", dataset_path,
             "--num_layers", str(request.model_params.get('num_layers', 12)),
             "--attention_heads", str(request.model_params.get('attention_heads', 1)),
             "--hidden_size", str(request.model_params.get('hidden_size', 64)),

             "python", TRAIN_MODEL_PATH,
             "--task", request.task,
             "--model_name", request.model_name,
+            "--dataset", os.path.abspath(dataset_path),
             "--num_layers", str(request.model_params.get('num_layers', 12)),
             "--attention_heads", str(request.model_params.get('attention_heads', 1)),
             "--hidden_size", str(request.model_params.get('hidden_size', 64)),