diff --git a/requirements-cpu.txt b/requirements-cpu.txt index e503987..8cd716e 100644 --- a/requirements-cpu.txt +++ b/requirements-cpu.txt @@ -1,6 +1,6 @@ torch==1.12.0+cpu torchvision==0.13.0+cpu -ray[serve]==2.0.0 +ray[serve]==2.2.0 fastapi==0.81.0 imageio==2.21.2 kornia==0.6.6 diff --git a/requirements-gpu.txt b/requirements-gpu.txt index 492895c..1963ca7 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -1,6 +1,6 @@ torch==1.12.0+cu116 torchvision==0.13.0+cu116 -ray[serve]==2.0.0 +ray[serve]==2.2.0 fastapi==0.81.0 imageio==2.21.2 kornia==0.6.6