Fix gs None error in legacy _ensure_stacked path

This commit is contained in:
2026-05-19 02:17:53 +00:00
parent 48fa64dfda
commit bab1f75f29

View File

@@ -231,10 +231,8 @@ class CuTeDSLMoERunner:
self._l2_scale_b = assemble_scales_3d_side(self.l2_sf)
self.l1_fp4 = None
self.l1_sf = None
self.l1_gs = None
self.l2_fp4 = None
self.l2_sf = None
self.l2_gs = None
self._l1_gsb = torch.tensor(self.l1_gs, dtype=torch.float32, device=self.device)
self._l2_gsb = torch.tensor(self.l2_gs, dtype=torch.float32, device=self.device)