add onnx to wd14 tagger

2026-04-18 01:30:02 +00:00 · 2023-10-08 20:31:10 +08:00
parent 2d87bb648f
commit 70fe7e18be
2 changed files with 47 additions and 12 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -19,8 +19,10 @@ huggingface-hub==0.15.1
 # requests==2.28.2
 # timm==0.6.12
 # fairscale==0.4.13
-# for WD14 captioning
+# for WD14 captioning (tensroflow or onnx)
 # tensorflow==2.10.1
+# onnx==1.14.1
+# onnxruntime==1.16.0
 # open clip for SDXL
 open-clip-torch==2.20.0
 # for kohya_ss library