bugfix for weight loading for qwen3-next (#21313)

2026-03-26 21:21:00 +08:00
parent 8febc3650d
commit 0906e45cec
1 changed files with 2 additions and 2 deletions
@@ -135,10 +135,10 @@ class Qwen3GatedDeltaNet(nn.Module):

        # Override weight_loader for packed checkpoint format.
        # Must capture original_loader BEFORE overwriting.
-        self.in_proj_qkvz.weight.weight_loader = self._make_packed_weight_loader(
+        self.in_proj_qkvz.weight._weight_loader = self._make_packed_weight_loader(
            self.in_proj_qkvz
        )
-        self.in_proj_ba.weight.weight_loader = self._make_packed_weight_loader(
+        self.in_proj_ba.weight._weight_loader = self._make_packed_weight_loader(
            self.in_proj_ba
        )