Update fine_tune.py

fine_tune.py

@@ -10,7 +10,7 @@ from transformers import (
 from peft import LoraConfig, PeftModel
 from trl import SFTTrainer
 from fastapi import FastAPI
-from pydantic import BaseModel # 1. ADD THIS IMPORT
+from pydantic import BaseModel
 import uvicorn
 
 # --- Configuration ---
@@ -69,7 +69,7 @@ if not os.path.exists(os.path.join(output_dir, 'adapter_config.json')):
         group_by_length=True,
         lr_scheduler_type="linear",
         push_to_hub=True,
-        hub_model_id="Nutnell/DirectEd-AI",
+        hub_model_id="Nutnell/direct-ed-finetune-job",
     )
 
     # Initialize Trainer
@@ -77,7 +77,7 @@ if not os.path.exists(os.path.join(output_dir, 'adapter_config.json')):
         model=model,
         train_dataset=dataset,
         peft_config=peft_config,
-        dataset_text_field="text", # Ensure your dataset has a 'text' column
+        dataset_text_field="text",
         args=training_arguments,
     )