Views
No views yet
1from unsloth import FastLanguageModel, SFTTrainer
2from datasets import load_dataset
3
4model, tokenizer = FastLanguageModel.from_pretrained(
5 "unsloth/Llama-3.2-3B-Instruct",
6 max_seq_length=2048,
7 load_in_4bit=True
8)
9
10dataset = load_dataset("mlabonne/FineTome-100k", split="train")
11dataset = dataset.map(lambda x: {"text": tokenizer.apply_chat_template(x["conversations"], tokenize=False)})
12
13trainer = SFTTrainer(model=model, tokenizer=tokenizer, train_dataset=dataset)
14trainer.train()
15
16---
17license: mit
18---