From 0a90b2abd327119b778a262696fe5bb36d9c4b38 Mon Sep 17 00:00:00 2001 From: Belem Zhang Date: Fri, 2 Aug 2024 15:26:37 +0800 Subject: [PATCH] Update safety_checker_int32_reduceSum.onnx --- demos/stable-diffusion-1.5/index.js | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demos/stable-diffusion-1.5/index.js b/demos/stable-diffusion-1.5/index.js index cbfbb4f..0ff809e 100644 --- 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,