LordNeel commited on
Commit
390b224
·
verified ·
1 Parent(s): 3cc883e

Upload train_glm47_flash_test.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. train_glm47_flash_test.py +1 -1
train_glm47_flash_test.py CHANGED
@@ -142,7 +142,7 @@ trainer = SFTTrainer(
142
  model=model,
143
  train_dataset=dataset,
144
  args=training_config,
145
- tokenizer=tokenizer,
146
  )
147
 
148
  print("\n" + "=" * 60)
 
142
  model=model,
143
  train_dataset=dataset,
144
  args=training_config,
145
+ processing_class=tokenizer,
146
  )
147
 
148
  print("\n" + "=" * 60)