|
1 | 1 | from .decoder_noise import DecoderNoise |
| 2 | +from .dynamic_conditioning import DynamicConditioning |
2 | 3 | from .easy_samplers import ( |
3 | 4 | LinearOverlapLatentTransition, |
4 | 5 | LTXVBaseSampler, |
5 | 6 | LTXVExtendSampler, |
6 | 7 | LTXVInContextSampler, |
7 | 8 | ) |
8 | | -from .film_grain import LTXVFilmGrain |
| 9 | +from .gemma_encoder import LTXVGemmaCLIPModelLoader, LTXVGemmaEnhancePrompt |
| 10 | +from .gemma_encoder_mgpu_loader import LTXVGemmaCLIPModelLoaderMGPU |
9 | 11 | from .guide import LTXVAddGuideAdvanced |
10 | | -from .latent_adain import LTXVAdainLatent |
11 | | -from .latent_upsampler import LTXVLatentUpsampler |
12 | | -from .latents import LTXVSelectLatents, LTXVSetVideoLatentNoiseMasks |
| 12 | +from .guiders import GuiderParametersNode, MultimodalGuiderNode |
| 13 | +from .latent_norm import ( |
| 14 | + LTXVAdainLatent, |
| 15 | + LTXVPerStepAdainPatcher, |
| 16 | + LTXVPerStepStatNormPatcher, |
| 17 | + LTXVStatNormLatent, |
| 18 | +) |
| 19 | +from .latents import ( |
| 20 | + LTXVAddLatentGuide, |
| 21 | + LTXVImgToVideoConditionOnly, |
| 22 | + LTXVSelectLatents, |
| 23 | + LTXVSetVideoLatentNoiseMasks, |
| 24 | +) |
13 | 25 | from .looping_sampler import LTXVLoopingSampler, MultiPromptProvider |
| 26 | +from .low_vram_loaders import ( |
| 27 | + LowVRAMAudioVAELoader, |
| 28 | + LowVRAMCheckpointLoader, |
| 29 | + LowVRAMLatentUpscaleModelLoader, |
| 30 | +) |
14 | 31 | from .masks import LTXVPreprocessMasks |
15 | 32 | from .nodes_registry import NODE_CLASS_MAPPINGS as RUNTIME_NODE_CLASS_MAPPINGS |
16 | 33 | from .nodes_registry import ( |
|
30 | 47 | from .tricks import NODE_CLASS_MAPPINGS as TRICKS_NODE_CLASS_MAPPINGS |
31 | 48 | from .tricks import NODE_DISPLAY_NAME_MAPPINGS as TRICKS_NODE_DISPLAY_NAME_MAPPINGS |
32 | 49 | from .utiltily_nodes import ImageToCPU |
33 | | -from .vae_patcher.vae_patcher import LTXVPatcherVAE |
| 50 | +from .vae_patcher import LTXVPatcherVAE |
34 | 51 |
|
35 | 52 | # Static node mappings, required for ComfyUI-Manager mapping to work |
36 | 53 | NODE_CLASS_MAPPINGS = { |
37 | 54 | "Set VAE Decoder Noise": DecoderNoise, |
38 | 55 | "LTXVLinearOverlapLatentTransition": LinearOverlapLatentTransition, |
39 | 56 | "LTXVAddGuideAdvanced": LTXVAddGuideAdvanced, |
| 57 | + "LTXVAddLatentGuide": LTXVAddLatentGuide, |
40 | 58 | "LTXVAdainLatent": LTXVAdainLatent, |
| 59 | + "LTXVImgToVideoConditionOnly": LTXVImgToVideoConditionOnly, |
| 60 | + "LTXVPerStepAdainPatcher": LTXVPerStepAdainPatcher, |
41 | 61 | "LTXVApplySTG": LTXVApplySTG, |
42 | 62 | "LTXVBaseSampler": LTXVBaseSampler, |
43 | 63 | "LTXVInContextSampler": LTXVInContextSampler, |
44 | 64 | "LTXVExtendSampler": LTXVExtendSampler, |
45 | | - "LTXVFilmGrain": LTXVFilmGrain, |
46 | 65 | "LTXVPreprocessMasks": LTXVPreprocessMasks, |
47 | | - "LTXVLatentUpsampler": LTXVLatentUpsampler, |
48 | 66 | "LTXVPatcherVAE": LTXVPatcherVAE, |
49 | 67 | "LTXVPromptEnhancer": LTXVPromptEnhancer, |
50 | 68 | "LTXVPromptEnhancerLoader": LTXVPromptEnhancerLoader, |
|
55 | 73 | "LTXVTiledSampler": LTXVTiledSampler, |
56 | 74 | "LTXVLoopingSampler": LTXVLoopingSampler, |
57 | 75 | "LTXVTiledVAEDecode": LTXVTiledVAEDecode, |
| 76 | + "MultimodalGuider": MultimodalGuiderNode, |
| 77 | + "GuiderParameters": GuiderParametersNode, |
58 | 78 | "STGAdvancedPresets": STGAdvancedPresetsNode, |
59 | 79 | "STGGuiderAdvanced": STGGuiderAdvancedNode, |
60 | 80 | "STGGuiderNode": STGGuiderNode, |
61 | 81 | "LTXVMultiPromptProvider": MultiPromptProvider, |
62 | 82 | "ImageToCPU": ImageToCPU, |
| 83 | + "LTXVStatNormLatent": LTXVStatNormLatent, |
| 84 | + "LTXVPerStepStatNormPatcher": LTXVPerStepStatNormPatcher, |
| 85 | + "LTXVGemmaCLIPModelLoader": LTXVGemmaCLIPModelLoader, |
| 86 | + "LTXVGemmaCLIPModelLoaderMGPU": LTXVGemmaCLIPModelLoaderMGPU, |
| 87 | + "LTXVGemmaEnhancePrompt": LTXVGemmaEnhancePrompt, |
| 88 | + "DynamicConditioning": DynamicConditioning, |
| 89 | + "LowVRAMCheckpointLoader": LowVRAMCheckpointLoader, |
| 90 | + "LowVRAMAudioVAELoader": LowVRAMAudioVAELoader, |
| 91 | + "LowVRAMLatentUpscaleModelLoader": LowVRAMLatentUpscaleModelLoader, |
63 | 92 | } |
64 | 93 |
|
65 | 94 | # Consistent display names between static and dynamic node mappings in nodes_registry.py, |
|
0 commit comments