Skip to content

Commit 373a64a

Browse files
authored
💬 Remove setting chat template in sft script (#4037)
1 parent 09e1924 commit 373a64a

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

trl/scripts/sft.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,6 @@
7777
SFTConfig,
7878
SFTTrainer,
7979
TrlParser,
80-
clone_chat_template,
8180
get_dataset,
8281
get_kbit_device_map,
8382
get_peft_config,
@@ -121,11 +120,6 @@ def main(script_args, training_args, model_args, dataset_args):
121120
model_args.model_name_or_path, trust_remote_code=model_args.trust_remote_code, use_fast=True
122121
)
123122

124-
# Set default chat template if needed
125-
if tokenizer.chat_template is None:
126-
# TODO: source should be passed as an argument
127-
model, tokenizer = clone_chat_template(model, tokenizer, "Qwen/Qwen3-0.6B")
128-
129123
# Load the dataset
130124
if dataset_args.datasets and script_args.dataset_name:
131125
logger.warning(

0 commit comments

Comments
 (0)