mlcommons / mobile_app_open

Mobile App Open
https://mlcommons.org/en/groups/inference-mobile/
Apache License 2.0
47 stars 23 forks source link

make the number of steps of stable diffusion configuration #921

Closed freedomtan closed 1 month ago

freedomtan commented 2 months ago

the tflite implementation https://github.com/mlcommons/mobile_app_open/blob/master/mobile_back_tflite/cpp/backend_tflite/stable_diffusion_pipeline.h#L40

supposed, both

should be configurable from the app or from the pbtxt (task description, https://github.com/mlcommons/mobile_app_open/blob/master/flutter/assets/tasks.pbtxt).

(let's change the tflite backend number of steps to 20 first).