⚗️📌 Try lower accelerate

Signed-off-by: Evaline Ju <[email protected]>
caikit · Feb 14, 2025 · a3fd545 · a3fd545
1 parent 9e77327
commit a3fd545
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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",
 ]