mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
fix openpose and add test-compile
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -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):
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user