Prepare Model+Dataset

python -c "
from transformers import AutoTokenizer, AutoModelForCausalLM
model_id = 'unsloth/DeepSeek-R1-Distill-Llama-8B'
AutoTokenizer.from_pretrained(model_id).save_pretrained('./models/deepseek')
AutoModelForCausalLM.from_pretrained(model_id).save_pretrained('./models/deepseek')
"

python -c "
from datasets import load_dataset
dataset = load_dataset('Sulav/mental_health_counseling_conversations_sharegpt', split='train')
dataset.save_to_disk('./data/sharegpt')
"

RUN

uv pip install

uv run train