pi-ux-ce/wd14-tagger-standalone
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Support gpu.

Browse Source
This commit is contained in:
corkborg
2024-04-23 01:48:39 +09:00
parent 282472f212
commit babad14542
4 changed files with 37 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
requirements.txt
+1 -2
View File
@@ -1,7 +1,6 @@
pandas==2.1.1
huggingface-hub==0.17.3
torch==2.0.1
Pillow==10.0.1
deepdanbooru==1.0.2
onnxruntime==1.16.0
onnxruntime==1.17.3
opencv-python==4.8.0.76