define before reference

pytorch · Apr 9, 2024 · 935df99 · 935df99
1 parent a90ab63
commit 935df99
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/quantize.py b/quantize.py
@@ -587,10 +587,9 @@ def create_quantized_state_dict(self):
             if isinstance(mod, torch.nn.Linear):
                 assert not mod.bias
                 in_features = mod.in_features
+                group_size = self.group_size
                 if group_size is None or group_size == 0:
                     group_size = in_features
-                else:
-                    group_size = self.group_size
                 # print("in features:", in_features, " out features:", out_features)
                 # assert out_features % 8 == 0, "require out_features % 8 == 0"
                 # print(f"linear: {fqn}, in={in_features}, out={out_features}")