remove g_idx arg (vllm-project#139)

markmc · Aug 22, 2024 · fe62831 · fe62831
1 parent 17aa177
commit fe62831
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/compressed_tensors/quantization/lifecycle/helpers.py b/src/compressed_tensors/quantization/lifecycle/helpers.py
@@ -64,7 +64,7 @@ def update_layer_weight_quant_params(
     if attached_weight is not None:
         weight = weight.to(attached_weight.dtype)
 
-    updated_scale, updated_zero_point = observer(weight, g_idx=g_idx)
+    updated_scale, updated_zero_point = observer(weight)
 
     # update scale and zero point
     device = next(layer.parameters()).device