FIX sft script print_trainable_parameters attr lookup (#1928)

anch0vy · web-flow · commit a019f8690d52 · 2024-07-15T17:09:14.000+02:00
diff --git a/examples/sft/train.py b/examples/sft/train.py
@@ -137,7 +137,7 @@ def main(model_args, data_args, training_args):
         max_seq_length=data_args.max_seq_length,
     )
     trainer.accelerator.print(f"{trainer.model}")
-    if hasattr(trainer.model, "print_trainable_parameters()"):
+    if hasattr(trainer.model, "print_trainable_parameters"):
         trainer.model.print_trainable_parameters()
 
     # train

Original file line number	Diff line number	Diff line change
`@@ -137,7 +137,7 @@ def main(model_args, data_args, training_args):`
`137`	`137`	`max_seq_length=data_args.max_seq_length,`
`138`	`138`	`)`
`139`	`139`	`trainer.accelerator.print(f"{trainer.model}")`
`140`		`- if hasattr(trainer.model, "print_trainable_parameters()"):`
	`140`	`+ if hasattr(trainer.model, "print_trainable_parameters"):`
`141`	`141`	`trainer.model.print_trainable_parameters()`
`142`	`142`
`143`	`143`	`# train`