Refactor SFT forward_backward to use TrainBatch#731

Open

HosseinKaviani-H wants to merge 5 commits into

meta-pytorch:mainfrom

HosseinKaviani-H:SFT_Re

Commits on Feb 2, 2026

Refactor SFT forward_backward to use TrainBatch
Hossein Kavianihamedani
committed
Refactor SFT collate to return TrainBatch directly
Hossein Kavianihamedani
committed
Refactor SFT collate to return TrainBatch directly
Hossein Kavianihamedani
committed
Remove backward compatibility for dict in utils.py
Hossein Kavianihamedani
committed
Change tests to include TrainBatch
Hossein Kavianihamedani
committed