diff --git a/test/apis/realtime/text-generator/app/requirements-cpu.txt b/test/apis/realtime/text-generator/app/requirements-cpu.txt index 91ec4ae257..472d61b123 100644 --- a/test/apis/realtime/text-generator/app/requirements-cpu.txt +++ b/test/apis/realtime/text-generator/app/requirements-cpu.txt @@ -2,4 +2,4 @@ uvicorn[standard] fastapi transformers==3.0.* -f https://download.pytorch.org/whl/torch_stable.html -torch==1.7.1+cpu +torch==1.13.1 diff --git a/test/apis/realtime/text-generator/app/requirements-gpu.txt b/test/apis/realtime/text-generator/app/requirements-gpu.txt index 9ad41ccd52..dfe5a10e5c 100644 --- a/test/apis/realtime/text-generator/app/requirements-gpu.txt +++ b/test/apis/realtime/text-generator/app/requirements-gpu.txt @@ -2,4 +2,4 @@ uvicorn[standard]==0.16.0 sentencepiece==0.1.94 fastapi transformers==3.0.* -torch==1.10.2 +torch==1.13.1