George Krupenchenkov commited on
Commit
9ca63ee
·
1 Parent(s): d20cd4b

add hw6 markups

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -99,8 +99,10 @@ def infer(
99
  }
100
 
101
  print("in infer 1")
 
102
 
103
- if controlnet_enable:
 
104
  print("in controlnet_enable")
105
  if control_mode == "canny":
106
  controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_canny",
 
99
  }
100
 
101
  print("in infer 1")
102
+ print("controlnet_enable", controlnet_enable)
103
 
104
+ controlnet = None
105
+ if controlnet_enable and control_image is not None:
106
  print("in controlnet_enable")
107
  if control_mode == "canny":
108
  controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_canny",