Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -387,9 +387,9 @@ def infer(
|
|
387 |
|
388 |
# #control_mode = "pose_estimation" # Режим работы ControlNet
|
389 |
|
390 |
-
print('control_mode = ', control_mode.
|
391 |
|
392 |
-
if control_mode.
|
393 |
|
394 |
# Инициализация ControlNet
|
395 |
controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch_dtype)
|
|
|
387 |
|
388 |
# #control_mode = "pose_estimation" # Режим работы ControlNet
|
389 |
|
390 |
+
print('control_mode = ', control_mode.value)
|
391 |
|
392 |
+
if control_mode.value == "pose_estimation":
|
393 |
|
394 |
# Инициализация ControlNet
|
395 |
controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch_dtype)
|