Skip to content

Commit ab70311

Browse files
committed
remove unneeded code
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent 8ae9004 commit ab70311

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

src/compressed_tensors/utils/offload.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -103,10 +103,6 @@ def get_execution_device(module: torch.nn.Module) -> torch.device:
103103
return module._hf_hook.execution_device
104104
device = next(module.parameters()).device
105105

106-
# offload only gets set for leaf modules, fallback to checking for device type
107-
if device.type == "meta":
108-
return module._hf_hook.execution_device
109-
110106
return device
111107

112108

0 commit comments

Comments
 (0)