Skip to content

Commit

Permalink
USING SD-1.5 INSTEAD OF REALISTIC VISION
Browse files Browse the repository at this point in the history
  • Loading branch information
usamaehsan committed Nov 18, 2023
1 parent 6458b60 commit cac5020
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -237,6 +237,7 @@ def build_pipe(
init_image= None
got_size= False
for name, [image, conditioning_scale, mask_image] in inputs.items():
print(name)
if image is None:
continue

Expand Down Expand Up @@ -407,12 +408,12 @@ def predict(
raise Exception("missing weights")

control_inputs= {
"lineart": [lineart_image, lineart_conditioning_scale, None],
"brightness": [brightness_image, brightness_conditioning_scale, None],
"tile": [tile_image, tile_conditioning_scale, None],
"lineart": [lineart_image, lineart_conditioning_scale, None],
"inpainting": [inpainting_image, inpainting_conditioning_scale, mask_image],
# "scribble": [scribble_image, scribble_conditioning_scale, None],
# "depth": [depth_image, depth_conditioning_scale, None],
"brightness": [brightness_image, brightness_conditioning_scale, None],
}
sorted_control_inputs= sort_dict_by_string(sorted_controlnets, control_inputs)

Expand Down

0 comments on commit cac5020

Please sign in to comment.