diff --git a/Dockerfile b/Dockerfile
index 4d089e3331d..c4ef451ec95 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM quay.io/opendatahub/text-generation-inference:fast-283ec87
+FROM quay.io/opendatahub/text-generation-inference:fast-58c9786
 
 USER root
 
diff --git a/caikit/Pipfile b/caikit/Pipfile
index f6304f55e83..ca8ad697de9 100644
--- a/caikit/Pipfile
+++ b/caikit/Pipfile
@@ -9,11 +9,11 @@ verify_ssl = true
 name = "pytorch_nightly"
 
 [packages]
-caikit-nlp = {editable = true, ref = "5f1d9cc27522a66bddc7e356b4f4bc39aa61e3f2", git = "https://github.com/opendatahub-io/caikit-nlp.git"}
+caikit-nlp = {editable = true, ref = "729c5ff9869b9c98865ad15609b95972f65e1dbc", git = "https://github.com/opendatahub-io/caikit-nlp.git"}
 torch = {version = "==2.1.0.dev20230730+cu118", index = "pytorch_nightly" }
 pytorch-triton = {version = "==2.1.0+e6216047b8", index = "pytorch_nightly"}
 
 [dev-packages]
 
 [requires]
-python_version = "3.9"
\ No newline at end of file
+python_version = "3.9"