Skip to content
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

use time step embedding from file #928

Merged
merged 29 commits into from
Jan 14, 2025
Merged
Changes from 1 commit
Commits
Show all changes
29 commits
Select commit Hold shift + click to select a range
c78b889
feat: pass task-specific config to backend (#922)
anhappdev Oct 15, 2024
fad36c1
feat: add ConvertOutputs() API (#927)
anhappdev Oct 23, 2024
48cb33c
feat: timestamp-embedding-parser (WIP)
RSMNYS Oct 28, 2024
6d9bc91
disabled bitcode to be able compile with new XCode
RSMNYS Oct 28, 2024
fadee08
chore: formatting
RSMNYS Oct 28, 2024
1bddf37
refactor: use custom setting in Core ML backend to detect NCHW input.…
anhappdev Nov 12, 2024
463d974
fix: resolve crash due to permission denied on Android Play Store ver…
anhappdev Nov 12, 2024
e0f6813
chore: increase Android minSdkVersion from 21 to 30 (#859)
anhappdev Nov 19, 2024
1008c49
feat: finalized SD pipeline to use embedding from the binary file.
RSMNYS Dec 2, 2024
aba5d7d
Merge branch 'master' into features/timestamp-embedding-parser
RSMNYS Dec 2, 2024
8527f36
refactor: updated embedding_utils to parse pkl file
RSMNYS Dec 3, 2024
1cec8b3
chore: linting
RSMNYS Dec 3, 2024
47c54fa
fix: fixed lint issue in neuron
RSMNYS Dec 3, 2024
1c97942
chore: BUILD cleanup
RSMNYS Dec 3, 2024
4b67590
chore: cleanup
RSMNYS Dec 3, 2024
8eb4f4a
chore: ignore .fvm
RSMNYS Dec 6, 2024
0835157
chore: updated model paths and checksums for stable diffusion benchma…
RSMNYS Dec 6, 2024
b035768
chore: reverse timesteps and embeddings to support descending order o…
RSMNYS Dec 29, 2024
f2f132a
chore: fixed formatting
RSMNYS Dec 31, 2024
b7ad4ee
Merge branch 'submission-v4.1' into features/timestamp-embedding-parser
RSMNYS Jan 3, 2025
336bd76
chore: added links to the sd models and timestep embeddings file
RSMNYS Jan 3, 2025
f222a14
chore: add the proper name for the embedding_timesteps file
RSMNYS Jan 6, 2025
b14424e
chore: added missed declaration for backend_convert_outputs
RSMNYS Jan 6, 2025
3e5a5e2
chore: clang formatting
RSMNYS Jan 6, 2025
809b11d
chore: added missed files
RSMNYS Jan 7, 2025
ddedee4
chore: fixed build file for the pixel backend
RSMNYS Jan 7, 2025
3d58e5a
chore: bazel formatting
RSMNYS Jan 7, 2025
6170627
fix: added missed interface implementation for pixel
RSMNYS Jan 7, 2025
f29dccd
chore: clang formatting
RSMNYS Jan 7, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
chore: cleanup
  • Loading branch information
RSMNYS committed Dec 3, 2024
commit 4b675901b71e86354eb5db1014fc97d2f1051468
4 changes: 2 additions & 2 deletions WORKSPACE
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
workspace(name = "mlperf_app")

load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
load("@bazel_tools//tools/build_defs/repo:git.bzl", "new_git_repository")
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")

http_archive(
name = "bazel_skylib",
@@ -49,11 +49,11 @@ http_archive(
],
)

load("@rules_python//python:repositories.bzl", "python_register_toolchains")
load(
"@org_tensorflow//tensorflow/tools/toolchains/python:python_repo.bzl",
"python_repository",
)
load("@rules_python//python:repositories.bzl", "python_register_toolchains")

python_repository(name = "python_version_repo")

4 changes: 2 additions & 2 deletions mobile_back_tflite/cpp/backend_tflite/BUILD
Original file line number Diff line number Diff line change
@@ -40,10 +40,10 @@ cc_library(
name = "embedding_utils",
srcs = ["embedding_utils.cc"],
hdrs = ["embedding_utils.h"],
visibility = ["//visibility:public"],
deps = [
"@org_tensorflow//tensorflow/core:tflite_portable_logging",
],
visibility = ["//visibility:public"],
)

cc_library(
@@ -79,8 +79,8 @@ cc_library(
"//conditions:default": [],
}),
deps = [
":tflite_settings",
":embedding_utils",
":tflite_settings",
"//flutter/cpp:utils",
"//flutter/cpp/c:headers",
"@org_tensorflow//tensorflow/core:tflite_portable_logging",
2 changes: 1 addition & 1 deletion mobile_back_tflite/cpp/backend_tflite/neuron/BUILD
Original file line number Diff line number Diff line change
@@ -72,8 +72,8 @@ cc_library(
":tflite_settings",
"//flutter/cpp:utils",
"//flutter/cpp/c:headers",
"//mobile_back_tflite/cpp/backend_tflite:tflite_settings",
"//mobile_back_tflite/cpp/backend_tflite:embedding_utils",
"//mobile_back_tflite/cpp/backend_tflite:tflite_settings",
"@org_tensorflow//tensorflow/core:tflite_portable_logging",
"@org_tensorflow//tensorflow/lite/c:c_api",
"@org_tensorflow//tensorflow/lite/c:common",
Original file line number Diff line number Diff line change
@@ -112,8 +112,7 @@ mlperf_backend_ptr_t StableDiffusionPipeline::backend_create(
}

std::string ts_embedding_path =
std::string(model_path) +
"/timestep_steps_20_int32_embedding_1x1280_float32.bin.ts";
std::string(model_path) + "/timestep_embeddings_data.pkl";
if (!EmbeddingManager::getInstance().load_timestep_embeddings(
ts_embedding_path)) {
LOG(ERROR) << "Failed to load timestep embeddings from "
Loading