Skip to content

Commit 94a0bf5

Browse files
committed
Merge remote-tracking branch 'origin' into kylesayrs/extend-hadamard
2 parents 249323c + 52e7074 commit 94a0bf5

File tree

2 files changed

+2
-4
lines changed
  • src/compressed_tensors

2 files changed

+2
-4
lines changed

src/compressed_tensors/quantization/lifecycle/apply.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -183,9 +183,7 @@ def apply_quantization_config(
183183
replace_module(model, name, compressed_linear)
184184

185185
# target matched - add layer and scheme to target list
186-
submodule.quantization_scheme = _scheme_from_targets(
187-
target_to_scheme, targets, name
188-
)
186+
submodule.quantization_scheme = scheme
189187

190188
names_to_scheme[name] = submodule.quantization_scheme
191189

src/compressed_tensors/transform/factory/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def _apply_to_module(self, module: Module, args: TransformArgs):
9797
:param args: defines how the transform will be applied to the target module
9898
"""
9999
# create transform as submodule
100-
transform_name = f"{self.name}_{args.location}"
100+
transform_name = f"{self.name}_{args.location.value}"
101101
transform = self.create_transform(module, args)
102102
register_offload_module(module, transform_name, transform) # (1)
103103

0 commit comments

Comments
 (0)