Skip to content

Commit 69c7378

Browse files
stduhpfleejet
andauthored
fix: force binary mask for inpaint models (#589)
Co-authored-by: leejet <leejet714@gmail.com>
1 parent 838beb9 commit 69c7378

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ggml_extend.hpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -383,8 +383,10 @@ __STATIC_INLINE__ void sd_apply_mask(struct ggml_tensor* image_data,
383383
for (int ix = 0; ix < width; ix++) {
384384
for (int iy = 0; iy < height; iy++) {
385385
float m = ggml_tensor_get_f32(mask, ix, iy);
386+
m = round(m); // inpaint models need binary masks
387+
ggml_tensor_set_f32(mask, m, ix, iy);
386388
for (int k = 0; k < channels; k++) {
387-
float value = ((float)(m < 254.5 / 255)) * (ggml_tensor_get_f32(image_data, ix, iy, k) - .5) + .5;
389+
float value = (1 - m) * (ggml_tensor_get_f32(image_data, ix, iy, k) - .5) + .5;
388390
ggml_tensor_set_f32(output, value, ix, iy, k);
389391
}
390392
}

0 commit comments

Comments
 (0)