fix: Hardcode Legacy behavior to True to resolve warning. (#446)

* Set Legacy to True Testing Legacy behavior set to True Signed-off-by: Luka Dojcinovic <56648891+Luka-D@users.noreply.github.com> * Changed legacy to False Signed-off-by: Luka Dojcinovic <56648891+Luka-D@users.noreply.github.com> * Set Legacy = True We've decided to maintain the previous legacy behavior, coding it here to avoid that warning. Signed-off-by: Luka Dojcinovic <56648891+Luka-D@users.noreply.github.com> --------- Signed-off-by: Luka Dojcinovic <56648891+Luka-D@users.noreply.github.com>
foundation-model-stack · Jan 30, 2025 · a6e4f9d · a6e4f9d
1 parent 3daf5bb
commit a6e4f9d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tuning/sft_trainer.py b/tuning/sft_trainer.py
@@ -233,6 +233,7 @@ def train(
         ),
         cache_dir=train_args.cache_dir,
         use_fast=True,
+        legacy=True,
     )
 
     # Calculate and save additional metrics to track later.