alextorelli commited on
Commit
7e453f6
·
verified ·
1 Parent(s): 244cb91

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -107,7 +107,7 @@ async def run_training_job(
107
 
108
  # Create AutoTrain config file (more reliable than CLI args)
109
  config = {
110
- "task": "llm:sft",
111
  "base_model": base_model,
112
  "project_name": f"codechef-{project_name}",
113
  "data_path": str(dataset_path),
 
107
 
108
  # Create AutoTrain config file (more reliable than CLI args)
109
  config = {
110
+ "task": "lm_training", # Changed from llm:sft
111
  "base_model": base_model,
112
  "project_name": f"codechef-{project_name}",
113
  "data_path": str(dataset_path),