32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from .final_frame_selector import (
|
|
FinalFrameSelector,
|
|
NODE_CLASS_MAPPINGS as FINAL_FRAME_SELECTOR_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as FINAL_FRAME_SELECTOR_DISPLAY_MAPPINGS
|
|
)
|
|
from .video_merge import (
|
|
VideoMerge,
|
|
NODE_CLASS_MAPPINGS as VIDEO_MERGE_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as VIDEO_MERGE_DISPLAY_MAPPINGS
|
|
)
|
|
from .youtube_video_downloader import (
|
|
YouTubeVideoDownloader,
|
|
NODE_CLASS_MAPPINGS as YOUTUBE_VIDEO_DOWNLOADER_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as YOUTUBE_VIDEO_DOWNLOADER_DISPLAY_MAPPINGS
|
|
)
|
|
from .video_prompt_generator import (
|
|
VideoPromptGenerator,
|
|
NODE_CLASS_MAPPINGS as VIDEO_PROMPT_GENERATOR_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as VIDEO_PROMPT_GENERATOR_DISPLAY_MAPPINGS
|
|
)
|
|
NODE_CLASS_MAPPINGS = {
|
|
**FINAL_FRAME_SELECTOR_MAPPINGS,
|
|
**VIDEO_MERGE_MAPPINGS,
|
|
**YOUTUBE_VIDEO_DOWNLOADER_MAPPINGS,
|
|
**VIDEO_PROMPT_GENERATOR_MAPPINGS
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**FINAL_FRAME_SELECTOR_DISPLAY_MAPPINGS,
|
|
**VIDEO_MERGE_DISPLAY_MAPPINGS,
|
|
**YOUTUBE_VIDEO_DOWNLOADER_DISPLAY_MAPPINGS,
|
|
**VIDEO_PROMPT_GENERATOR_DISPLAY_MAPPINGS
|
|
} |