diff --git a/lumigator/jobs/inference/requirements.txt b/lumigator/jobs/inference/requirements.txt index f72dee186..ff2539cf6 100644 --- a/lumigator/jobs/inference/requirements.txt +++ b/lumigator/jobs/inference/requirements.txt @@ -9,5 +9,5 @@ pydantic==2.10.6 python-box==7.3.2 s3fs==2024.2.0 sentencepiece==0.2.0 -torch==2.6.0 +torch==2.7.1+cpu transformers==4.49.0 diff --git a/lumigator/jobs/inference/requirements_cpu.txt b/lumigator/jobs/inference/requirements_cpu.txt index ac167a71f..f091acf1c 100644 --- a/lumigator/jobs/inference/requirements_cpu.txt +++ b/lumigator/jobs/inference/requirements_cpu.txt @@ -10,5 +10,5 @@ pydantic==2.10.6 python-box==7.3.2 s3fs==2024.2.0 sentencepiece==0.2.0 -torch==2.6.0 +torch==2.7.1+cpu transformers==4.49.0