Skip to content

Commit 8a72d60

Browse files
fix var name (#1235)
1 parent dd292ef commit 8a72d60

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

gptqmodel/nn_modules/qlinear/torch.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def __init__(
7171
if self.group_size != self.in_features:
7272
self.padded_infeatures = self.in_features + (-self.in_features % self.group_size)
7373
else:
74-
self.padded_infeatures = self.infeatures
74+
self.padded_infeatures = self.in_features
7575

7676
if self.bits in [2, 4, 8]:
7777
self.wf = torch.tensor(list(range(0, self.pack_dtype_bits, self.bits)), dtype=torch.int32).unsqueeze(0)

gptqmodel/nn_modules/qlinear/tritonv2.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def __init__(self, bits: int, group_size: int, desc_act: bool, sym: bool, in_fea
9090
if self.group_size != self.in_features:
9191
self.padded_infeatures = self.in_features + (-self.in_features % self.group_size)
9292
else:
93-
self.padded_infeatures = self.infeatures
93+
self.padded_infeatures = self.in_features
9494

9595
@classmethod
9696
def validate(cls, **args) -> Tuple[bool, Optional[Exception]]:

0 commit comments

Comments
 (0)