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 486d41b commit 3c0f767Copy full SHA for 3c0f767
plugins/mamba-cp/src/fms_acceleration_mcp/utils/utils.py
@@ -106,7 +106,6 @@ def patch_mamba_layers_with_cp_head(
106
device = model.device
107
for layer in tqdm(model.model.layers, desc="Swapping mamba layers"):
108
if hasattr(layer, "mamba") and layer.mamba is not None:
109
- print("mamba layer found")
110
mamba_layer = Mamba2CPHF(**config_ssm, **cp_args)
111
mamba_layer.load_state_dict(layer.mamba.state_dict())
112
setattr(layer, "mamba", mamba_layer)
0 commit comments