diff --git a/nodes.py b/nodes.py index 28f5a40..7ca5422 100644 --- a/nodes.py +++ b/nodes.py @@ -23,7 +23,7 @@ from .src.models.unet_3d import UNet3DConditionModel from .src.models.main_diffuser import AADiffusion -ROOT_PATH = os.path.join(comfy_paths.get_folder_paths("custom_nodes")[0], "./ComfyUI-AnimateAnyone-Evolved") +ROOT_PATH = os.path.dirname(os.path.abspath(__file__)) DEFAULT_CONFIG_PATH = os.path.join(ROOT_PATH, "./configs/default.yaml") CONFIG = OmegaConf.load(DEFAULT_CONFIG_PATH) @@ -328,4 +328,4 @@ def pose_guider_encode(self, pose_guider, pose_images): #print(f"pose_cond_tensor.shape: {pose_cond_tensor.shape}\pose_latent.shape: {pose_latent.shape}") #pose_cond_tensor.shape: torch.Size([1, 3, 24, 768, 512]) pose_latent.shape: torch.Size([1, 320, 24, 96, 64]) - return (pose_latent,) \ No newline at end of file + return (pose_latent,)