Update requirements.txt
should improve the performance by 30%
This commit is contained in:
committed by
GitHub
parent
42687f5bd9
commit
f0fae811d8
@@ -1,4 +1,4 @@
|
|||||||
--extra-index-url https://download.pytorch.org/whl/cu118
|
--extra-index-url https://download.pytorch.org/whl/cu128
|
||||||
|
|
||||||
numpy>=1.23.5,<2
|
numpy>=1.23.5,<2
|
||||||
typing-extensions>=4.8.0
|
typing-extensions>=4.8.0
|
||||||
@@ -10,12 +10,12 @@ psutil==5.9.8
|
|||||||
tk==0.1.0
|
tk==0.1.0
|
||||||
customtkinter==5.2.2
|
customtkinter==5.2.2
|
||||||
pillow==11.1.0
|
pillow==11.1.0
|
||||||
torch==2.5.1+cu118; sys_platform != 'darwin'
|
torch; sys_platform != 'darwin'
|
||||||
torch==2.5.1; sys_platform == 'darwin'
|
torch==2.5.1; sys_platform == 'darwin'
|
||||||
torchvision==0.20.1; sys_platform != 'darwin'
|
torchvision; sys_platform != 'darwin'
|
||||||
torchvision==0.20.1; sys_platform == 'darwin'
|
torchvision==0.20.1; sys_platform == 'darwin'
|
||||||
onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
|
onnxruntime-silicon==1.21.0; sys_platform == 'darwin' and platform_machine == 'arm64'
|
||||||
onnxruntime-gpu==1.17; sys_platform != 'darwin'
|
onnxruntime-gpu==1.21.0; sys_platform != 'darwin'
|
||||||
tensorflow; sys_platform != 'darwin'
|
tensorflow; sys_platform != 'darwin'
|
||||||
opennsfw2==0.10.2
|
opennsfw2==0.10.2
|
||||||
protobuf==4.23.2
|
protobuf==4.23.2
|
||||||
|
|||||||
Reference in New Issue
Block a user