fix openpose and add test-compile

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-06-02 17:06:18 +02:00
parent 3c2a742f40
commit 35aa8c6422
12 changed files with 215 additions and 49 deletions
+1 -4
View File
@@ -11,10 +11,7 @@ from diffusers.configuration_utils import ConfigMixin, register_to_config
from diffusers.models.modeling_utils import ModelMixin
from diffusers.models.autoencoders.vae import DecoderOutput
from diffusers.models.attention_processor import Attention, AttnProcessor
try:
from diffusers.models.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block
except Exception:
from diffusers.models.unets.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block
from diffusers.models.unets.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block
def zero_module(module):
+2 -7
View File
@@ -9,13 +9,8 @@ checked_ok = False
def check_dependencies():
global checked_ok # pylint: disable=global-statement
from installer import installed, install
packages = [
('ligo-segments', 'ligo-segments'),
]
for pkg in packages:
if not installed(pkg[1], quiet=True):
install(pkg[0], pkg[1], ignore=False)
from installer import install
install('ligo-segments')
try:
from ligo.segments import segment # pylint: disable=unused-import
checked_ok = True