84 lines
3.3 KiB
Python
84 lines
3.3 KiB
Python
# Compatibility re-export for older imports that expect funpack.py.
|
|
|
|
try:
|
|
from . import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
from .conditioning import (
|
|
FunPackGemmaEmbeddingRefiner,
|
|
FunPackLorebookEnhancer,
|
|
FunPackPromptCombiner,
|
|
FunPackPromptEnhancer,
|
|
FunPackStoryWriter,
|
|
)
|
|
from .image_processing import (
|
|
FunPackContinueVideo,
|
|
FunPackStoryMemKeyframeExtractor,
|
|
FunPackStoryMemLastFrameExtractor,
|
|
FunPackVideoStitch,
|
|
)
|
|
from .model_management import FunPackApplyLoraWeights, FunPackLoraLoader
|
|
from .samplers import FunPackHybridEuler2SSampler, sample_funpack_hybrid_euler_2s
|
|
except ImportError:
|
|
from conditioning import (
|
|
FunPackGemmaEmbeddingRefiner,
|
|
FunPackLorebookEnhancer,
|
|
FunPackPromptCombiner,
|
|
FunPackPromptEnhancer,
|
|
FunPackStoryWriter,
|
|
)
|
|
from image_processing import (
|
|
FunPackContinueVideo,
|
|
FunPackStoryMemKeyframeExtractor,
|
|
FunPackStoryMemLastFrameExtractor,
|
|
FunPackVideoStitch,
|
|
)
|
|
from model_management import FunPackApplyLoraWeights, FunPackLoraLoader
|
|
from samplers import FunPackHybridEuler2SSampler, sample_funpack_hybrid_euler_2s
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FunPackPromptCombiner": FunPackPromptCombiner,
|
|
"FunPackStoryMemKeyframeExtractor": FunPackStoryMemKeyframeExtractor,
|
|
"FunPackStoryMemLastFrameExtractor": FunPackStoryMemLastFrameExtractor,
|
|
"FunPackPromptEnhancer": FunPackPromptEnhancer,
|
|
"FunPackStoryWriter": FunPackStoryWriter,
|
|
"FunPackVideoStitch": FunPackVideoStitch,
|
|
"FunPackContinueVideo": FunPackContinueVideo,
|
|
"FunPackLorebookEnhancer": FunPackLorebookEnhancer,
|
|
"FunPackGemmaEmbeddingRefiner": FunPackGemmaEmbeddingRefiner,
|
|
"FunPackHybridEuler2SSampler": FunPackHybridEuler2SSampler,
|
|
"FunPackApplyLoraWeights": FunPackApplyLoraWeights,
|
|
"FunPackLoraLoader": FunPackLoraLoader,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FunPackPromptCombiner": "FunPack Prompt Combiner",
|
|
"FunPackStoryMemKeyframeExtractor": "FunPack StoryMem Keyframe Extractor",
|
|
"FunPackStoryMemLastFrameExtractor": "FunPack StoryMem Last Frame Extractor",
|
|
"FunPackPromptEnhancer": "FunPack Prompt Enhancer (Standalone)",
|
|
"FunPackStoryWriter": "FunPack Story Writer",
|
|
"FunPackVideoStitch": "FunPack Video Stitch",
|
|
"FunPackContinueVideo": "FunPack Continue Video",
|
|
"FunPackLorebookEnhancer": "FunPack Lorebook Enhancer",
|
|
"FunPackGemmaEmbeddingRefiner": "FunPack Gemma Embedding Refiner (Self-Refinement)",
|
|
"FunPackHybridEuler2SSampler": "FunPack Hybrid Euler 2S Sampler",
|
|
"FunPackApplyLoraWeights": "FunPack Apply LoRA Weights",
|
|
"FunPackLoraLoader": "FunPack LoRA Loader",
|
|
}
|
|
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
"FunPackGemmaEmbeddingRefiner",
|
|
"FunPackHybridEuler2SSampler",
|
|
"FunPackPromptCombiner",
|
|
"FunPackLorebookEnhancer",
|
|
"FunPackPromptEnhancer",
|
|
"FunPackStoryWriter",
|
|
"FunPackVideoStitch",
|
|
"FunPackContinueVideo",
|
|
"FunPackStoryMemKeyframeExtractor",
|
|
"FunPackStoryMemLastFrameExtractor",
|
|
"FunPackApplyLoraWeights",
|
|
"FunPackLoraLoader",
|
|
"sample_funpack_hybrid_euler_2s",
|
|
]
|