diff --git a/requirements/constraints-alfworld-vllm-cu124.txt b/requirements/constraints-alfworld-vllm-cu124.txt index fe397c1..1bec416 100644 --- a/requirements/constraints-alfworld-vllm-cu124.txt +++ b/requirements/constraints-alfworld-vllm-cu124.txt @@ -1,4 +1,4 @@ -torch==2.6.0+cu124 +torch==2.12.0 torchvision==0.21.0+cu124 torchaudio==2.6.0+cu124 transformers==4.57.3 diff --git a/requirements/constraints-search-vllm-cu124.txt b/requirements/constraints-search-vllm-cu124.txt index bc7abee..f1388a9 100644 --- a/requirements/constraints-search-vllm-cu124.txt +++ b/requirements/constraints-search-vllm-cu124.txt @@ -1,4 +1,4 @@ -torch==2.6.0+cu124 +torch==2.12.0 torchvision==0.21.0+cu124 torchaudio==2.6.0+cu124 transformers==4.57.3