Update train.py

train.py

@@ -4,7 +4,7 @@ from datasets import load_dataset
 from peft import LoraConfig, get_peft_model
 
 # Model & Tokenizer
-MODEL_NAME = "TinyLlama/TinyLlama-1.1B"  # Adjust if using your own model
+MODEL_NAME = "vv876803/tinyllama-victo"  # Adjust if using your own model
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16)