-
Notifications
You must be signed in to change notification settings - Fork 26
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.
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
feat: add new benchmark Stable Diffusion #903
Conversation
MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅ |
@RSMNYS You should be able to integrate the Stable Diffusion pipeline for mobile_back_tflite based on this branch. model_path: "local:///path/to/folder" then the model_path will be a folder path in: mlperf_backend_ptr_t mlperf_backend_create(const char* model_path, mlperf_backend_configuration_t* configs, const char* native_lib_path) |
my old resize() with TFLite resize op (https://github.com/tensorflow/tensorflow/blob/2ca3e974e350664c0e18b2ff5a52867490a15222/tensorflow/lite/examples/label_image/bitmap_helpers_impl.h#L30) |
@freedomtan to check this implementation |
Hi guys! I've pushed the code partially here: https://github.com/mlcommons/mobile_app_open/tree/features/pipeline_backend The idea was to create the pipeline (stable_diffusion_backend.cc). And initiate it from the dart-run-benchmark.cc, by checking the dataset type for example. Then in issueQuery we can process the stable diffusion steps. |
@RSMNYS I refactored the TFLite backend to support multiple pipelines. |
* feat: added stable diffusion pipeline (WIP) * - enabled stable diffusion pipeline - code formatting * style: code formatting * style: code formatting * fix: adjusted inputs and outputs for the stable_diffusion_pipeline * fix: adjusted sd flow * fix: added unconditional tokens generation * chore: adjustments
* Suggestions for few changes in coco_gen dataset * Fixing a path
…on' into anh/add-benchmark-stable-diffusion
* Dump output of SD model to file * Add Python script to convert RGB8 to PNG * Dump raw output only in accuracy mode
error C2039: '__fs': is not a member of 'std'
error: 'path' is unavailable: introduced in iOS 13.0
error: 'path' is unavailable: introduced in iOS 13.0
|
We can merge this PR into master now. I disabled the tests for |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. @mohitmundhragithub if you are OK with this. Let's merge it.
Relevant files can be downloaded from here:
https://drive.google.com/drive/folders/10zCF7_ctUIM7xVPyhw5P60utnvWVUcSy?usp=sharing