diff --git a/dbgpt_hub/predict/predict.py b/dbgpt_hub/predict/predict.py
index e4653b7..6d67722 100644
--- a/dbgpt_hub/predict/predict.py
+++ b/dbgpt_hub/predict/predict.py
@@ -49,7 +49,10 @@ def main():
 
     with open(predict_output_dir_name, "w") as f:
         for p in result:
-            f.write(p.replace("\n", " ") + "\n")
+            try:
+                f.write(p.replace("\n", " ") + "\n")
+            except:
+                f.write("Invalid Output!\n")
 
 
 if __name__ == "__main__":
diff --git a/poetry.lock b/poetry.lock
index b5298cd..e15138d 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -262,13 +262,13 @@ tests-no-zope = ["cloudpickle", "hypothesis", "mypy (>=1.1.1)", "pympler", "pyte
 
 [[package]]
 name = "bitsandbytes"
-version = "0.39.0"
+version = "0.41.2"
 description = "k-bit optimizers and matrix multiplication routines."
 optional = false
 python-versions = "*"
 files = [
-    {file = "bitsandbytes-0.39.0-py3-none-any.whl", hash = "sha256:69e79cfce501cad5e34e9d785f5caf1768ead64f3e9f93b6bac1753e04dfa39a"},
-    {file = "bitsandbytes-0.39.0.tar.gz", hash = "sha256:b453dacf854e3624fe72b1306b1a6916de4a5edfd88fa01bf98ba73b2e3030d4"},
+    {file = "bitsandbytes-0.41.2-py3-none-any.whl", hash = "sha256:5a2280761dc11c7a23a1be948cfd6a849c2e718012ee34316b979eb6c5634de2"},
+    {file = "bitsandbytes-0.41.2.tar.gz", hash = "sha256:787c14b63cc559e1b344f683497a9353ac2e256a3fe89972f960e7c428d5cce7"},
 ]
 
 [[package]]
@@ -4247,4 +4247,4 @@ multidict = ">=4.0"
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.10,<3.13"
-content-hash = "c604a299b6b9fcb011c963bc8470c4a758baba2bb8e04c17918928d5c7c70043"
+content-hash = "fd8272ceaaac7b87bea6c6c49f11a7a3a791dece4cfb7eac34578a6a561dba6f"
diff --git a/pyproject.toml b/pyproject.toml
index 93a6bd4..a415cb9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,7 +11,6 @@ python = ">=3.10,<3.13"
 transformers = "4.33.2"
 peft = "0.4.0"
 accelerate = "0.21.0"
-bitsandbytes = "0.39.0"
 einops = "0.6.1"
 evaluate = "0.4.0"
 scikit-learn = "1.2.2"
@@ -57,6 +56,7 @@ nvidia-nccl-cu12 = "2.18.1"
 nvidia-nvtx-cu12 = "12.1.105"
 triton = "2.1.0"
 nvidia-nvjitlink-cu12 = "^12.3.52"
+bitsandbytes = "0.41.2"
 
 [build-system]
 requires = ["poetry-core"]