chore: cleanup

mlcommons · Dec 3, 2024 · 4b67590 · 4b67590
1 parent 1c97942
commit 4b67590
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 7 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -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")
 

diff --git a/mobile_back_tflite/cpp/backend_tflite/BUILD b/mobile_back_tflite/cpp/backend_tflite/BUILD
@@ -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",

diff --git a/mobile_back_tflite/cpp/backend_tflite/neuron/BUILD b/mobile_back_tflite/cpp/backend_tflite/neuron/BUILD
@@ -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",

diff --git a/mobile_back_tflite/cpp/backend_tflite/stable_diffusion_pipeline.cc b/mobile_back_tflite/cpp/backend_tflite/stable_diffusion_pipeline.cc
@@ -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 "