Skip to content

Commit 88a7542

Browse files
delete original kernel slightly earlier
1 parent c39de82 commit 88a7542

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras/src/quantizers/gptq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -453,11 +453,11 @@ def quantize_and_correct_layer(
453453
quantized, self.original_layer.quantized_kernel.dtype
454454
)
455455

456+
del self.original_layer._kernel
456457
self.original_layer.quantized_kernel.assign(quantized)
457458
self.original_layer.kernel_scale.assign(scale)
458459
self.original_layer.kernel_zero.assign(zero)
459460
self.original_layer.g_idx.assign(g_idx)
460-
del self.original_layer._kernel
461461
self.original_layer.gptq = True
462462

463463
def free(self):

0 commit comments

Comments
 (0)