Update model TinyLlama/TinyLlama_v1.1

train.py

@@ -5,7 +5,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, Trainer, TrainingA
 dataset = load_dataset("pathii/css_design_snippets")
 
 # Load pre-trained model and tokenizer
-model_name = "deepseek-ai/DeepSeek-V3"
+model_name = "TinyLlama/TinyLlama_v1.1"
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)