Revert "Fix different device type adjustment in PP" (#8141)

2025-07-18 01:12:11 -07:00
parent 9d33fcfb8e
commit d0510f08fe
4 changed files with 27 additions and 25 deletions
@@ -144,7 +144,6 @@ class TpModelWorker:
            self.tp_size * self.pp_rank + tp_rank,
            self.world_group.cpu_group,
            src=self.world_group.ranks[0],
-            device=self.device,
        )[0]
        set_random_seed(self.random_seed)