Merge pull request #24 from ibelem/safety-checker-sd-1.5

Update Safety Checker model for reduceSum with int32 input for Stable Diffusion 1.5
microsoft · Aug 5, 2024 · 4445eac · 4445eac
2 parents 0126183 + 0a90b2a
commit 4445eac
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
@@ -719,7 +719,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,