Skip to content

Commit d852324

Browse files
Update examples/trl-integration.py
Co-authored-by: Quentin Gallouédec <45557362+qgallouedec@users.noreply.github.com>
1 parent 4c1ab0d commit d852324

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

examples/trl-integration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
dataset = Dataset.from_dict({"prompt": prompts, "completion": completions})
3636

3737
trainer = SFTTrainer(
38-
model="trl-internal-testing/tiny-LlamaForCausalLM-3",
38+
model="Qwen/Qwen3-0.6B",
3939
args=SFTConfig(
4040
output_dir="./model_output",
4141
num_train_epochs=1,

0 commit comments

Comments
 (0)