提交 cb5fb903 authored 作者: kijai's avatar kijai

switch to relative imports

上级 e90ac6dd
import os import os
import torch import torch
import sys
import numpy as np import numpy as np
import gc import gc
...@@ -13,13 +12,12 @@ from diffusers.schedulers import EulerDiscreteScheduler ...@@ -13,13 +12,12 @@ from diffusers.schedulers import EulerDiscreteScheduler
from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
script_directory = os.path.dirname(os.path.abspath(__file__)) script_directory = os.path.dirname(os.path.abspath(__file__))
sys.path.append(script_directory)
from mimicmotion.pipelines.pipeline_mimicmotion import MimicMotionPipeline from .mimicmotion.pipelines.pipeline_mimicmotion import MimicMotionPipeline
from mimicmotion.modules.unet import UNetSpatioTemporalConditionModel from .mimicmotion.modules.unet import UNetSpatioTemporalConditionModel
from mimicmotion.modules.pose_net import PoseNet from .mimicmotion.modules.pose_net import PoseNet
from lcm_scheduler import AnimateLCMSVDStochasticIterativeScheduler from .lcm_scheduler import AnimateLCMSVDStochasticIterativeScheduler
class MimicMotionModel(torch.nn.Module): class MimicMotionModel(torch.nn.Module):
def __init__(self, base_model_path, lcm=False): def __init__(self, base_model_path, lcm=False):
...@@ -247,8 +245,8 @@ class MimicMotionGetPoses: ...@@ -247,8 +245,8 @@ class MimicMotionGetPoses:
def process(self, ref_image, pose_images, include_body, include_hand, include_face): def process(self, ref_image, pose_images, include_body, include_hand, include_face):
device = mm.get_torch_device() device = mm.get_torch_device()
from mimicmotion.dwpose.util import draw_pose from .mimicmotion.dwpose.util import draw_pose
from mimicmotion.dwpose.dwpose_detector import DWposeDetector from .mimicmotion.dwpose.dwpose_detector import DWposeDetector
yolo_model = "yolox_l.onnx" yolo_model = "yolox_l.onnx"
dw_pose_model = "dw-ll_ucoco_384.onnx" dw_pose_model = "dw-ll_ucoco_384.onnx"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论