dezzman commited on
Commit
96bf16a
·
verified ·
1 Parent(s): c1e556a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -168,6 +168,8 @@ def infer(
168
  pipeline_mgr.get_hed_detector()
169
  )
170
  params['controlnet_conditioning_scale'] = float(cn_strength)
 
 
171
 
172
  if ip_enable:
173
  pipe.load_ip_adapter(IP_ADAPTER, subfolder="models", weight_name=IP_ADAPTER_WEIGHT_NAME)
 
168
  pipeline_mgr.get_hed_detector()
169
  )
170
  params['controlnet_conditioning_scale'] = float(cn_strength)
171
+ else:
172
+ params['image'] = torch.zeros((1, 3, 512, 512)).to(device) # заглушка, чтобы pipeline не падал
173
 
174
  if ip_enable:
175
  pipe.load_ip_adapter(IP_ADAPTER, subfolder="models", weight_name=IP_ADAPTER_WEIGHT_NAME)