Merge pull request #11 from ibelem/safety-checker-int32-reducesum

Update Safety Checker model for reduceSum with int32 input
microsoft · Aug 1, 2024 · 0126183 · 0126183
2 parents c52e0c4 + 10725ca
commit 0126183
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);