Bump to torch 2.7 (#77)

Co-authored-by: Daniel King <43149077+dakinggg@users.noreply.github.com>

Author: bowenyang008

.github/workflows/pr-cpu.yaml

@@ -22,8 +22,8 @@ jobs:
     strategy:
       matrix:
         include:
-        - name: "cpu-2.6.0"
-          container: mosaicml/pytorch:2.6.0_cpu-python3.11-ubuntu22.04
+        - name: "cpu-2.7.0"
+          container: mosaicml/pytorch:2.7.0_cpu-python3.12-ubuntu22.04
           markers: "not gpu and not only_release"
           pip_deps: "[cpu]"
           pytest_command: "coverage run -m pytest"

.github/workflows/pr-gpu.yaml

@@ -22,8 +22,8 @@ jobs:
       fail-fast: false
       matrix:
         include:
-        - name: "gpu-2.6.0-1"
-          container: mosaicml/llm-foundry:2.6.0_cu124-latest
+        - name: "gpu-2.7.0-1"
+          container: mosaicml/llm-foundry:2.7.0_cu128-latest
           markers: "gpu"
           pip_deps: "[gpu]"
           pytest_command: "coverage run -m pytest"
@@ -51,8 +51,8 @@ jobs:
       fail-fast: false
       matrix:
         include:
-        - name: "gpu-2.6.0-2"
-          container: mosaicml/llm-foundry:2.6.0_cu124-latest
+        - name: "gpu-2.7.0-2"
+          container: mosaicml/llm-foundry:2.7.0_cu128-latest
           markers: "gpu"
           pip_deps: "[gpu]"
           pytest_command: "coverage run -m pytest"

pyproject.toml

@@ -8,34 +8,34 @@ packages = { find = { include = ["compose_rl*"] } }
 
 [project]
 name = "compose_rl"
-version = "0.4.0"
+version = "0.7.0.dev0"
 dependencies = [
     'spacy==3.7.5',
     'spacy-alignments',
     'ray<3',
     'kubernetes==29.0.0',
-    'vllm>=0.8.3,<0.9.0',
+    'vllm>=0.9.0,<0.10.0',
     # for MATH dataset
     'omegaconf==2.4.0.dev3', # needed for antrl4-4.11 package
-    'sympy==1.13.1', # this version is only compatible with antlr-4.11 package
+    'sympy==1.13.3', # this version is only compatible with antlr-4.11 package
     'antlr4-python3-runtime==4.11', # this is the version compatible with minieval
 ]
 
 [project.optional-dependencies]
 cpu = [
-    'llm-foundry[all-cpu]@git+https://github.com/mosaicml/llm-foundry.git@a51c96cbd280b76fa6bf45fd1f12e48f3c99e8e6#egg=llmfoundry',
+    'llm-foundry[all-cpu]@git+https://github.com/mosaicml/llm-foundry.git@main#egg=llmfoundry',
 ]
 gpu = [
-    'llm-foundry[all]@git+https://github.com/mosaicml/llm-foundry.git@a51c96cbd280b76fa6bf45fd1f12e48f3c99e8e6#egg=llmfoundry',
+    'llm-foundry[all]@git+https://github.com/mosaicml/llm-foundry.git@main#egg=llmfoundry',
 ]
 dev = [
-    'llm-foundry[dev]@git+https://github.com/mosaicml/llm-foundry.git@a51c96cbd280b76fa6bf45fd1f12e48f3c99e8e6#egg=llmfoundry',
+    'llm-foundry[dev]@git+https://github.com/mosaicml/llm-foundry.git@main#egg=llmfoundry',
 ]
 released = [
-    'llm-foundry[all]>=0.17.1,<0.21.0',
+    'llm-foundry[all]>=0.21.0',
 ]
 cpu_released = [
-    'llm-foundry[all-cpu]>=0.17.1,<0.21.0',
+    'llm-foundry[all-cpu]>=0.21.0',
 ]
 
 # Registry entry points