fix: Remove outdated TODO in convrsion script
Branch: BambaArchitecture Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
This commit is contained in:
parent
204e78fba1
commit
97e6ba8d99
1 changed files with 0 additions and 1 deletions
|
@ -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"""
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue