search

pushpendra910 / Stable-Diffusion

This repo contains code for the stable diffusion model and its conversion to tensorflow lite models.
Other
4 stars 1 forks source link

Does tflite support Android GPU or npu? #1

Open ZTMIDGO opened 1 year ago

ZTMIDGO commented 1 year ago

Hello, I am preparing to port stable diffusion to Android to run with TensorFlow, but I have a question, does the current tflite support GPU or npu, I have run stable diffusion on Android using onnxruntime, unfortunately it does not Support GPU or npu.

this is my project: https://github.com/ZTMIDGO/Android-Stable-diffusion-ONNX

pushpendra910 commented 1 year ago

You can try TensorFlow Lite GPU delegate and Vendor-specific libraries. Delegates are better options to try.