Skip to content

Commit 6c3e938

Browse files
committed
chore: dead code clean-up
1 parent edd1bbf commit 6c3e938

1 file changed

Lines changed: 0 additions & 21 deletions

File tree

wrappers.py

Lines changed: 0 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -117,27 +117,6 @@ def override(self, *args, virtual_vram_gb=4.0, donor_device="cpu",
117117
inner_model._distorch_v2_meta = {"full_allocation": full_allocation}
118118

119119
logger.info(f"[MultiGPU DisTorch V2] Full allocation string: {full_allocation}")
120-
logger.mgpu_mm_log(f"[MODEL_SETUP] Setting DisTorch model properties: virtual_vram_gb={virtual_vram_gb}")
121-
122-
if hasattr(out[0], 'model'):
123-
mp = out[0]
124-
mp_id = id(mp)
125-
inner_model = getattr(mp, 'model', None)
126-
inner_model_id = id(inner_model) if inner_model else None
127-
inner_model_name = type(inner_model).__name__ if inner_model else "None"
128-
inner_id_str = f"0x{inner_model_id:x}" if inner_model_id is not None else "None"
129-
130-
logger.mgpu_mm_log(f"[OBJECT_CHAIN_SET] ModelPatcher: mp_id=0x{mp_id:x}, inner_model_id={inner_id_str}, inner_model_type={inner_model_name}")
131-
132-
elif hasattr(out[0], 'patcher') and hasattr(out[0].patcher, 'model'):
133-
mp = out[0].patcher
134-
mp_id = id(mp)
135-
inner_model = getattr(mp, 'model', None)
136-
inner_model_id = id(inner_model) if inner_model else None
137-
inner_model_name = type(inner_model).__name__ if inner_model else "None"
138-
inner_id_str = f"0x{inner_model_id:x}" if inner_model_id is not None else "None"
139-
140-
logger.mgpu_mm_log(f"[OBJECT_CHAIN_SET] ModelPatcher via patcher: mp_id=0x{mp_id:x}, inner_model_id={inner_id_str}, inner_model_type={inner_model_name}")
141120

142121
return out
143122

0 commit comments

Comments
 (0)