From 97e6ba8d99c89389b6b769bedea7304e28793a2c Mon Sep 17 00:00:00 2001
From: Gabe Goodhart <ghart@us.ibm.com>
Date: Thu, 12 Dec 2024 15:02:05 -0700
Subject: [PATCH] fix: Remove outdated TODO in convrsion script

Branch: BambaArchitecture

Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
---
 convert_hf_to_gguf.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py
index 0bf9b0fbc..ece57527f 100755
--- a/convert_hf_to_gguf.py
+++ b/convert_hf_to_gguf.py
@@ -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"""