Skip to content

Commit 1039100

Browse files
committed
prepend input hooks
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 34f1343 commit 1039100

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
@@ -105,7 +105,7 @@ def input_hook(_, args):
105105
input = args[0]
106106
return transform(input)
107107

108-
module.register_forward_pre_hook(input_hook)
108+
module.register_forward_pre_hook(input_hook, prepend=True)
109109

110110
# eagerly apply transformation to weight
111111
elif args.location in (

0 commit comments

Comments
 (0)