23 lines
809 B
Python
23 lines
809 B
Python
import folder_paths
|
|
from .animatediff.logger import logger
|
|
from .animatediff.utils_model import get_available_motion_models, Folders
|
|
from .animatediff.model_injection import prepare_dinklink_register_definitions
|
|
from .animatediff.motion_module_ad import prepare_dinklink_motion_module_ad
|
|
from .animatediff.nodes import AnimateDiffExtension
|
|
from .animatediff.dinklink import init_dinklink
|
|
|
|
if len(get_available_motion_models()) == 0:
|
|
logger.error(f"No motion models found. Please download one and place in: {folder_paths.get_folder_paths(Folders.ANIMATEDIFF_MODELS)}")
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ["WEB_DIRECTORY"]
|
|
|
|
init_dinklink()
|
|
prepare_dinklink_register_definitions()
|
|
prepare_dinklink_motion_module_ad()
|
|
|
|
|
|
async def comfy_entrypoint() -> AnimateDiffExtension:
|
|
return AnimateDiffExtension()
|