chore: Switch WD14 captioning deps to ONNX Runtime from Tensorflow (#73)

Author: aoirint

README.md

@@ -95,7 +95,7 @@ sudo docker run --rm --gpus all \
   --v_parameterization
 ```
 
-### Example: WD14 Captioning (Tensorflow)
+### Example: WD14 Captioning (ONNX)
 
 ```shell
 mkdir -p "./cache/wd14_tagger_model_cache"
@@ -110,5 +110,6 @@ sudo docker run --rm --gpus all \
   aoirint/sd_scripts \
   finetune/tag_images_by_wd14_tagger.py \
   --model_dir "/wd14_tagger_model_cache/wd14_tagger_model" \
+  --onnx \
   /work/my_dataset-20230715.1/img
 ```

pyproject.toml

@@ -31,8 +31,9 @@ dependencies = [
   "requests==2.32.4",
   "timm==0.6.12",
   "fairscale==0.4.13",
-  # for WD14 captioning (tensorflow)
-  "tensorflow==2.12.1",
+  # for WD14 captioning (onnx)
+  "onnx==1.15.0",
+  "onnxruntime-gpu==1.17.1",
   # open clip for SDXL
   "open-clip-torch==2.20.0",
   # For logging