fix state dict adapter in forge engine (#1563)

This was broken by #1526, updating to match the changes in train.py.
Tested via a run in forge

Author: ebsmothers

torchtitan/experiments/forge/engine.py

@@ -216,7 +216,9 @@ def __init__(self, job_config: ForgeJobConfig):
             states={"train_state": self},
             checkpoint_config=job_config.checkpoint,
             sd_adapter=(
-                self.train_spec.state_dict_adapter(model_args)
+                self.train_spec.state_dict_adapter(
+                    model_args, job_config.model.hf_assets_path
+                )
                 if self.train_spec.state_dict_adapter
                 else None
             ),