diff --git a/nodes.py b/nodes.py index d587f36..8a40617 100644 --- a/nodes.py +++ b/nodes.py @@ -1,6 +1,5 @@ import os import torch -import sys import numpy as np import gc @@ -13,13 +12,12 @@ from diffusers.schedulers import EulerDiscreteScheduler from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection script_directory = os.path.dirname(os.path.abspath(__file__)) -sys.path.append(script_directory) -from mimicmotion.pipelines.pipeline_mimicmotion import MimicMotionPipeline -from mimicmotion.modules.unet import UNetSpatioTemporalConditionModel -from mimicmotion.modules.pose_net import PoseNet +from .mimicmotion.pipelines.pipeline_mimicmotion import MimicMotionPipeline +from .mimicmotion.modules.unet import UNetSpatioTemporalConditionModel +from .mimicmotion.modules.pose_net import PoseNet -from lcm_scheduler import AnimateLCMSVDStochasticIterativeScheduler +from .lcm_scheduler import AnimateLCMSVDStochasticIterativeScheduler class MimicMotionModel(torch.nn.Module): def __init__(self, base_model_path, lcm=False): @@ -247,8 +245,8 @@ class MimicMotionGetPoses: def process(self, ref_image, pose_images, include_body, include_hand, include_face): device = mm.get_torch_device() - from mimicmotion.dwpose.util import draw_pose - from mimicmotion.dwpose.dwpose_detector import DWposeDetector + from .mimicmotion.dwpose.util import draw_pose + from .mimicmotion.dwpose.dwpose_detector import DWposeDetector yolo_model = "yolox_l.onnx" dw_pose_model = "dw-ll_ucoco_384.onnx"