Skip to content

Commit ca600ca

Browse files
committed
fix typos
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent ab70311 commit ca600ca

File tree

3 files changed

+2
-3
lines changed

3 files changed

+2
-3
lines changed

src/compressed_tensors/compressors/quantized_compressors/base.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,6 @@ def compress(
115115
)
116116

117117
# update state dict
118-
del model_state[name]
119118
for key, value in compressed_values.items():
120119
compressed_dict[prefix + key] = value.to(save_device)
121120

tests/test_compressors/quantized_compressors/test_fp8_quant.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ def test_reload_match(
158158
model.dummy.weight,
159159
scale=model.dummy.weight_scale,
160160
zero_point=model.dummy.weight_zero_point,
161-
args=module_name_to_scheme["dummy"].weight,
161+
args=module_name_to_scheme["dummy"].weights,
162162
)
163163
assert torch.equal(fake_quant_dummy, reconstructed_dense["dummy"].get("weight"))
164164

tests/test_compressors/quantized_compressors/test_int_quant.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ def test_reload_match(strategy, group_size, sc, zp, tmp_path):
153153
dense_state_dict["dummy2.weight"],
154154
scale=dense_state_dict["dummy2.weight_scale"],
155155
zero_point=dense_state_dict["dummy2.weight_zero_point"],
156-
args=module_name_to_scheme["dummy2"].weight,
156+
args=module_name_to_scheme["dummy2"].weights,
157157
)
158158
assert torch.equal(
159159
fake_quant_dummy2, reconstructed_dense["dummy2"].get("weight").to(torch.float32)

0 commit comments

Comments
 (0)