Skip to content

Commit adae77b

Browse files
authored
[Transform] [Bugfix] Fix enum value serialization in python>=3.11 (#350)
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent 4b81ac7 commit adae77b

File tree

1 file changed

+1
-1
lines changed
  • src/compressed_tensors/transform/factory

1 file changed

+1
-1
lines changed

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)