From 10725ca7c351269995cfcb38b6157a5afdbb1d47 Mon Sep 17 00:00:00 2001
From: Belem Zhang <belem.zhang@intel.com>
Date: Fri, 14 Jun 2024 13:12:20 +0800
Subject: [PATCH] Update Safety Checker model for reduceSum with int32 input

---
 demos/sd-turbo/index.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demos/sd-turbo/index.js b/demos/sd-turbo/index.js
index bb04fc5..8e0957f 100644
--- a/demos/sd-turbo/index.js
+++ b/demos/sd-turbo/index.js
@@ -238,7 +238,7 @@ async function load_models(models) {
         modelUrl = `${config.model}/${name}/model.onnx`;
       } else if (name == "safety_checker") {
         modelNameInLog = "Safety Checker";
-        modelUrl = `${config.model}/${name}/model.onnx`;
+        modelUrl = `${config.model}/${name}/safety_checker_int32_reduceSum.onnx`;
       }
       log(`[Load] Loading model ${modelNameInLog} · ${model.size}`);
       let modelBuffer = await getModelOPFS(`sd_turbo_${name}`, modelUrl, false);