New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add OpenVINO and ROCm ONNX Runtime for WD14 #1213

Merged

kohya-ss merged 1 commit into kohya-ss:dev from Disty0:dev

Mar 27, 2024

Contributor

Disty0 commented Mar 27, 2024 •

edited

Loading

Add ONNX Runtime OpenVINO for any vendor on any platform. (Tested on Intel ARC A770 and AMD RX 7900 XTX)
OpenVINO will default to GPU.1 if multiple GPUs are available.
Add ONNX Runtime ROCm for AMD on Linux. (Tested on AMD RX 7900 XTX)
Update IPEX libs.

For OpenVINO installation: pip install onnxruntime-openvino
For ROCm installation: https://rocm.docs.amd.com/projects/radeon/en/latest/docs/install/install-onnx.html


          Add OpenVINO and ROCm ONNX Runtime for WD14

6f7e93d

Owner

kohya-ss commented Mar 27, 2024

Thank you for updating WD14 tagger! It is really helpful.

kohya-ss merged commit b86af67 into kohya-ss:dev

1 check passed

Disty0 mentioned this pull request

AMD ROCm installer bmaltais/kohya_ss#2167

Merged

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet