We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2bf33c8 commit 33b71b3Copy full SHA for 33b71b3
src/compressed_tensors/transform/apply.py
@@ -62,7 +62,6 @@ def _tie_offloaded_tensors(model: torch.nn.Module):
62
# map from offloaded tensor pointers to module-key locations
63
offloaded_ptrs: dict[int, List[Tuple[torch.nn.Module, str]]] = defaultdict(list)
64
for module in model.modules():
65
- # NOTE: previously asserted that parent._hf_hook.place_submodules=False
66
if has_offloaded_params(module):
67
for key, _ in module.named_parameters(recurse=False):
68
param = module._hf_hook.weights_map[key]
0 commit comments