fix: Remove outdated TODO in convrsion script

Branch: BambaArchitecture

Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
This commit is contained in:
Gabe Goodhart 2024-12-12 15:02:05 -07:00
parent 204e78fba1
commit 97e6ba8d99

View file

@ -3093,7 +3093,6 @@ class Mamba2Model(Model):
return data_torch.squeeze()
# TODO: Switch to BambaForCausalLM once ready in transformers
@Model.register("BambaForCausalLM")
class BambaModel(Mamba2Model):
"""Bamba is a hybrid SSM + Attention model that uses Mamba2 SSM layers"""