Update safety_checker_int32_reduceSum.onnx

microsoft · Aug 2, 2024 · 0a90b2a · 0a90b2a
1 parent 0126183
commit 0a90b2a
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,