Spaces:

vishaljoshi24
/

trl-4-dnd

Paused

App Files Files Community

trl-4-dnd / quickstart.py

vishaljoshi24's picture

Initial Commit

a080fe0 3 days ago

218 Bytes

	from trl import SFTTrainer
	from datasets import load_dataset

	dataset = load_dataset("trl-lib/Capybara", split="train")

	trainer = SFTTrainer(
	model="Qwen/Qwen2.5-0.5B",
	train_dataset=dataset,
	)
	trainer.train()