diff --git a/caikit_nlp/modules/text_generation/text_generation_local.py b/caikit_nlp/modules/text_generation/text_generation_local.py
index 8ffc7e4f..3261f910 100644
--- a/caikit_nlp/modules/text_generation/text_generation_local.py
+++ b/caikit_nlp/modules/text_generation/text_generation_local.py
@@ -648,8 +648,8 @@ def _launch_training(
         print("Dir trainer!", dir(trainer))
         print("Trainer dict", trainer.__dict__)
 
-        # Confirmed stuck in training
-        assert False
+        # # Confirmed stuck in training
+        # assert False
 
         # Start training via Trainer.train function
         trainer.train()
diff --git a/pyproject.toml b/pyproject.toml
index 02c0349b..cc3d9cb2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,7 +20,7 @@ dependencies = [
     "grpcio>=1.62.2", # explicitly pin grpc dependencies to a recent version to avoid pip backtracking
     "grpcio-reflection>=1.62.2",
     "grpcio-health-checking>=1.62.2",
-    "accelerate>=0.22.0",
+    "accelerate>=0.22.0,<1",
     "datasets>=2.4.0",
     "huggingface-hub",
     "numpy>=1.22.4",
@@ -31,7 +31,7 @@ dependencies = [
     "tokenizers>=0.13.3",
     "torch>=2.3.1,<2.6.0",
     "tqdm>=4.65.0",
-    "transformers>=4.44.0,<4.47.0",
+    "transformers>=4.44.0,<4.49.0",
     "peft>=0.13.0,<0.15.0",
 ]