diff --git a/demos/stable-diffusion-1.5/index.js b/demos/stable-diffusion-1.5/index.js index a18bf99..7c7e8e2 100644 --- 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,