DeepSeek-R1-Distill-Qwen-1.5B Fine-Tuned on GSM8K with Chain-of-Thought Augmentation
Model Overview
This model is a fine-tuned version of DeepSeek-R1-Distill-Qwen-1.5B, trained on the OpenAI GSM8K dataset, augmented with Chain-of-Thought (CoT) reasoning using DeepSeek-V3. The fine-tuning process enhances the model’s mathematical problem-solving abilities, allowing it to provide step-by-step solutions with deeper reasoning.
🔹 Key Features
Base Model: DeepSeek-R1-Distill-Qwen-1.5B
Fine-Tuned On: GSM8K dataset with DeepSeek-V3-enhanced reasoning
You can load the model with Hugging Face Transformers:
python
1from transformers import AutoModelForCausalLM, AutoTokenizer
23model_name ="eagle0504/deepseek-r1-qwen-1.5b-gsm8k-enhanced"45tokenizer = AutoTokenizer.from_pretrained(model_name)6model = AutoModelForCausalLM.from_pretrained(model_name)78prompt ="A farmer has 24 apples. He gives 6 to each of his 3 children. How many does he have left?"9inputs = tokenizer(prompt, return_tensors="pt")10output = model.generate(**inputs, max_length=200)11print(tokenizer.decode(output[0], skip_special_tokens=True))
🔬 Expected Performance
Compared to the base DeepSeek-R1-Distill-Qwen-1.5B, this fine-tuned model:
Provides more detailed Chain-of-Thought (CoT) explanations for GSM8K problems.
Improves logical reasoning and step-by-step answer formulation.
Generates clearer, more structured solutions, making it ideal for educational use.