Files
Endless-Nodes/batchers/__init__.py
tusharbhutt 91ac2eb06c Add files via upload
Adding Flux Kontext functionality
2025-07-07 13:46:47 -06:00

47 lines
1.8 KiB
Python

"""
EndlessSeaofStars Custom Nodes for ComfyUI
Batch processing nodes with specialized support for FLUX and SDXL models
"""
from .endless_batchers import (
EndlessNode_SimpleBatchPrompts,
EndlessNode_FluxBatchPrompts,
EndlessNode_SDXLBatchPrompts,
EndlessNode_BatchNegativePrompts,
EndlessNode_PromptCounter,
EndlessNode_FluxKontextBatchPrompts,
EndlessNode_ReplicateLatents,
# IGNORE ME, I AM NOT READY!!
# from .endless_fluxlatent import (
# EndlessNode_FluxLatentReplicator,
# EndlessNode_FluxLatentReplicatorFromPrompts,
)
# Node class mappings for ComfyUI
NODE_CLASS_MAPPINGS = {
"SimpleBatchPrompts": EndlessNode_SimpleBatchPrompts,
"FluxBatchPrompts": EndlessNode_FluxBatchPrompts,
"SDXLBatchPrompts": EndlessNode_SDXLBatchPrompts,
"BatchNegativePrompts": EndlessNode_BatchNegativePrompts,
"PromptCounter": EndlessNode_PromptCounter,
"FluxKontextBatchPrompts": EndlessNode_FluxKontextBatchPrompts,
"EndlessReplicateLatents": EndlessNode_ReplicateLatents,
# IGNORE ME, I AM NOT READY!!
# "LatentReplicator": EndlessNode_FluxLatentReplicator,
# "LatentReplicatorPrompts": EndlessNode_FluxLatentReplicatorFromPrompts,
}
# Display names for ComfyUI interface
NODE_DISPLAY_NAME_MAPPINGS = {
"SimpleBatchPrompts": "Simple Batch Prompts",
"FluxBatchPrompts": "FLUX Batch Prompts",
"SDXLBatchPrompts": "SDXL Batch Prompts",
"BatchNegativePrompts": "Batch Negative Prompts",
"PromptCounter": "Prompt Counter",
"FluxKontextBatchPrompts": "FLUX Kontext Batch Prompts",
"EndlessReplicateLatents": "Replicate Latents",
# IGNORE ME, I AM NOT READY!!
# "LatentReplicator": "Latent Replicator",
# "LatentReplicatorPrompts": "Latent Replicator from Prompts",
}