Merge pull request #29 from microsoft/revert-27-revert-24-safety-chec…

…ker-sd-1.5 Revert "Revert Update Safety Checker model for reduceSum with int32 input for Stable Diffusion 1.5
microsoft · Aug 13, 2024 · 89ad288 · 89ad288
2 parents 4c80f1b + ca5c7ff
commit 89ad288
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demos/stable-diffusion-1.5/index.js b/demos/stable-diffusion-1.5/index.js
@@ -713,7 +713,7 @@ async function loadModel(modelName /*:String*/, executionProvider /*:String*/) {
     //  Outputs:
     //    float16 out_images
     //    bool has_nsfw_concepts
-    modelPath = Utils.modelPath() + "safety-checker.onnx";
+    modelPath = Utils.modelPath() + "safety_checker_int32_reduceSum.onnx";
     freeDimensionOverrides = {
       batch: 1,
       channels: 3,