stmasson commited on
Commit
e388b5a
·
verified ·
1 Parent(s): 0d48090

Upload scripts/train_n8n_sft.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. scripts/train_n8n_sft.py +1 -1
scripts/train_n8n_sft.py CHANGED
@@ -257,7 +257,7 @@ trainer = SFTTrainer(
257
  train_dataset=train_dataset,
258
  eval_dataset=val_dataset,
259
  peft_config=lora_config,
260
- tokenizer=tokenizer,
261
  )
262
 
263
  # Afficher les parametres entrainables
 
257
  train_dataset=train_dataset,
258
  eval_dataset=val_dataset,
259
  peft_config=lora_config,
260
+ processing_class=tokenizer, # renamed from tokenizer in TRL 0.12+
261
  )
262
 
263
  # Afficher les parametres entrainables