Fixes

pyproject.toml

@@ -14,7 +14,7 @@ requires-python = ">=3.8"
 license = { file = "LICENSE" }
 dependencies = [
     "numpy",
-    "torch>=2.0",
+    "torch>=2.0,<2.2",
     "omegaconf",
     "rich",
     "boto3",

scripts/beaker/beaker_interactive.sh

@@ -49,4 +49,4 @@ gh repo clone allenai/LLM
 cd LLM
 
 # Install other dependencies.
-pip install -e '.[dev]'
+pip install -e '.[all]'

scripts/test_entrypoint.sh

@@ -29,7 +29,7 @@ git checkout --quiet "$COMMIT_SHA"
 
 # Install dependencies.
 pip install --upgrade pip
-pip install --no-cache-dir '.[dev]'
+pip install --no-cache-dir '.[all]'
 
 # Create directory for results.
 mkdir -p /results