Skip to content

Commit 501d8f0

Browse files
authored
Update quant_utils.py
Signed-off-by: Chenjie Luo <[email protected]>
1 parent 76ce70a commit 501d8f0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modelopt/torch/export/quant_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -878,7 +878,7 @@ def postprocess_state_dict(state_dict: dict, maxbound: float, quantization: str
878878
and value.shape[0] == 1
879879
):
880880
post_state_dict[key] = value.squeeze(0)
881-
881+
882882
# remove real quant parameters from the state dict
883883
keys_to_delete = []
884884
for key, value in post_state_dict.items():

0 commit comments

Comments
 (0)