Skip to content

Commit ce81753

Browse files
committed
fix: enable scribble mode with mask input by default (AUTOMATIC1111#7)
1 parent 1004973 commit ce81753

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

README.md

+2
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@ Thanks & Inspired: kohya-ss/sd-webui-additional-networks
1717

1818
### Install
1919

20+
Some users may need to install the cv2 library before using it: `pip install opencv-python`
21+
2022
Install prettytable if you want to use img2seg preprocessor: `pip install prettytable`
2123

2224
1. Open "Extensions" tab.

scripts/controlnet.py

+1
Original file line numberDiff line numberDiff line change
@@ -258,6 +258,7 @@ def restore_networks():
258258
if not ((image['mask'][:, :, 0]==0).all() or (image['mask'][:, :, 0]==255).all()):
259259
print("using mask as input")
260260
input_image = HWC3(image['mask'][:, :, 0])
261+
scribble_mode = True
261262

262263
if scribble_mode:
263264
detected_map = np.zeros_like(input_image, dtype=np.uint8)

0 commit comments

Comments
 (0)