rename modules folder
This commit is contained in:
+2
-2
@@ -1,7 +1,7 @@
|
||||
# Stuff
|
||||
|
||||
articulate_module/models/*.pth
|
||||
spline_module/models/*.tar
|
||||
module_articulate/models/*.pth
|
||||
module_spline/models/*.tar
|
||||
checkpoints/*
|
||||
!checkpoints/place_checkpoints_here.txt
|
||||
|
||||
|
||||
+4
-4
@@ -99,11 +99,11 @@ seg_model_dict = {
|
||||
"vox-adv": seg10_segment_names,
|
||||
}
|
||||
|
||||
ARTICULATE_MODEL_PATH = "articulate_module/models/vox256.pth"
|
||||
ARTICULATE_CFG_PATH = "articulate_module/config/vox256.yaml"
|
||||
ARTICULATE_MODEL_PATH = "module_articulate/models/vox256.pth"
|
||||
ARTICULATE_CFG_PATH = "module_articulate/config/vox256.yaml"
|
||||
|
||||
SPLINE_MODES = ['relative', 'standard', 'avd']
|
||||
SPLINE_DEFAULT = 'relative'
|
||||
|
||||
SPLINE_MODEL_PATH = "spline_module/models/vox.pth.tar"
|
||||
SPLINE_CFG_PATH = "spline_module/config/vox-256.yaml"
|
||||
SPLINE_MODEL_PATH = "module_spline/models/vox.pth.tar"
|
||||
SPLINE_CFG_PATH = "module_spline/config/vox-256.yaml"
|
||||
@@ -5,10 +5,10 @@ import yaml
|
||||
from comfy.utils import ProgressBar
|
||||
from scipy.spatial import ConvexHull
|
||||
|
||||
from .articulate_module.avd_network import AVDNetwork
|
||||
from .articulate_module.bg_motion_predictor import BGMotionPredictor
|
||||
from .articulate_module.generator import Generator
|
||||
from .articulate_module.region_predictor import RegionPredictor
|
||||
from .module_articulate.avd_network import AVDNetwork
|
||||
from .module_articulate.bg_motion_predictor import BGMotionPredictor
|
||||
from .module_articulate.generator import Generator
|
||||
from .module_articulate.region_predictor import RegionPredictor
|
||||
from .sync_batchnorm.replicate import DataParallelWithCallback
|
||||
|
||||
|
||||
|
||||
+4
-4
@@ -7,10 +7,10 @@ from comfy.utils import ProgressBar
|
||||
|
||||
from .inference_fomm import find_best_frame
|
||||
|
||||
from .spline_module.avd_network import AVDNetwork
|
||||
from .spline_module.dense_motion import DenseMotionNetwork
|
||||
from .spline_module.keypoint_detector import KPDetector
|
||||
from .spline_module.inpainting_network import InpaintingNetwork
|
||||
from .module_spline.avd_network import AVDNetwork
|
||||
from .module_spline.dense_motion import DenseMotionNetwork
|
||||
from .module_spline.keypoint_detector import KPDetector
|
||||
from .module_spline.inpainting_network import InpaintingNetwork
|
||||
|
||||
|
||||
def spline_inference(
|
||||
|
||||
Reference in New Issue
Block a user