Update train.py

train.py

@@ -1,4 +1,11 @@
-import os import zipfile import pandas as pd import torch from datasets import Dataset from transformers import TrainingArguments from unsloth import FastLanguageModel from trl import SFTTrainer
+import os
+import zipfile
+import pandas as pd
+import torch from datasets
+import Dataset from transformers
+import TrainingArguments from unsloth
+import FastLanguageModel from trl
+import SFTTrainer
 
 def main(): # 1) Load Unsloth 4-bit model + tokenizer model_name = "HuggingFaceTB/SmolLM2-1.7B" max_seq_length = 2048 dtype = torch.float16 load_in_4bit = True
 
@@ -104,7 +111,7 @@ tokenizer.save_pretrained(final_dir)
 print(f"Saved final adapter to {final_dir}")
 
 # 10) Zip the final model
-zip_path = "model_final.zip"
+zip_path = "model.zip"
 print(f"Zipping model directory {final_dir} to {zip_path}...")
 with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as z:
     for root, _, files in os.walk(final_dir):