Skip to content

Commit 7f11b97

Browse files
committed
change name
Signed-off-by: Kyle Sayers <[email protected]>
1 parent fa0f793 commit 7f11b97

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/llmcompressor/transformers/sparsification/compressed_tensors_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
CompressionFormat,
1212
ModelCompressor,
1313
SparsityCompressionConfig,
14-
is_module_offloaded,
14+
has_offloaded_params,
1515
update_offload_parameter,
1616
)
1717
from loguru import logger
@@ -162,7 +162,7 @@ def patch_tied_tensors_bug(model: torch.nn.Module):
162162

163163
if storage_ptr(input_embed.weight) == storage_ptr(output_embed.weight):
164164
for module in (input_embed, output_embed):
165-
if not is_module_offloaded(module):
165+
if not has_offloaded_params(module):
166166
# create new storage ptr for onloaded weight
167167
untied_data = module.weight.data.clone()
168168
module.weight.data = untied_data

0 commit comments

Comments
 (0)