Skip to content

Commit 3c0f767

Browse files
committed
fix: remove print stmts
Signed-off-by: Mehant Kammakomati <[email protected]>
1 parent 486d41b commit 3c0f767

File tree

1 file changed

+0
-1
lines changed
  • plugins/mamba-cp/src/fms_acceleration_mcp/utils

1 file changed

+0
-1
lines changed

plugins/mamba-cp/src/fms_acceleration_mcp/utils/utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,6 @@ def patch_mamba_layers_with_cp_head(
106106
device = model.device
107107
for layer in tqdm(model.model.layers, desc="Swapping mamba layers"):
108108
if hasattr(layer, "mamba") and layer.mamba is not None:
109-
print("mamba layer found")
110109
mamba_layer = Mamba2CPHF(**config_ssm, **cp_args)
111110
mamba_layer.load_state_dict(layer.mamba.state_dict())
112111
setattr(layer, "mamba", mamba_layer)

0 commit comments

Comments
 (0)