fix dependency

kohya-ss · Oct 8, 2023 · d6f458f · d6f458f
1 parent b8b8402
commit d6f458f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -22,6 +22,7 @@ huggingface-hub==0.15.1
 # for WD14 captioning (tensroflow or onnx)
 # tensorflow==2.10.1
 # onnx==1.14.1
+# onnxruntime-gpu==1.16.0
 # onnxruntime==1.16.0
 # open clip for SDXL
 open-clip-torch==2.20.0