Skip to content

Commit 49ac13c

Browse files
committed
Reconstruct to empty tensor
1 parent 8ff65b8 commit 49ac13c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

exllamav3/modules/quant/exl3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ def get_weight_tensor(self):
131131

132132

133133
def get_inner_weight_tensor(self):
134-
w = torch.zeros((self.in_features, self.out_features), dtype = torch.half, device = self.trellis.device)
134+
w = torch.empty((self.in_features, self.out_features), dtype = torch.half, device = self.trellis.device)
135135
ext.reconstruct(w, self.trellis, self.K, self.mcg_mult, self.mul1_mult)
136136
return w
137137

0 commit comments

Comments
 (0)