Skip to content

Commit 4445eac

Browse files
authoredAug 5, 2024··
Merge pull request #24 from ibelem/safety-checker-sd-1.5
Update Safety Checker model for reduceSum with int32 input for Stable Diffusion 1.5
2 parents 0126183 + 0a90b2a commit 4445eac

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
 

‎demos/stable-diffusion-1.5/index.js

+1-1
Original file line numberDiff line numberDiff line change
@@ -719,7 +719,7 @@ async function loadModel(modelName /*:String*/, executionProvider /*:String*/) {
719719
// Outputs:
720720
// float16 out_images
721721
// bool has_nsfw_concepts
722-
modelPath = Utils.modelPath() + "safety-checker.onnx";
722+
modelPath = Utils.modelPath() + "safety_checker_int32_reduceSum.onnx";
723723
freeDimensionOverrides = {
724724
batch: 1,
725725
channels: 3,

0 commit comments

Comments
 (0)
Please sign in to comment.