Hugging Face's logo

Spaces:
vishaljoshi24
/
trl-4-dnd
Paused

App Files Community
vishaljoshi24 commited on
Commit
49e8a0b
·
1 Parent(s): bc05830

SFT with gpt2

Browse files
Files changed (1) hide show
  1. quickstart.py +1 -1
quickstart.py CHANGED
@@ -4,7 +4,7 @@ from datasets import load_dataset
4
  dataset = load_dataset("trl-lib/Capybara", split="train")
5
 
6
  trainer = SFTTrainer(
7
- model="Qwen/Qwen2.5-0.5B",
8
  train_dataset=dataset,
9
  )
10
  trainer.train()
 
4
  dataset = load_dataset("trl-lib/Capybara", split="train")
5
 
6
  trainer = SFTTrainer(
7
+ model="openai-community/gpt2",
8
  train_dataset=dataset,
9
  )
10
  trainer.train()