Vito Sansevero
c5e00c130b
- Checkpoint Folder Iterator loads one checkpoint per queued run from a folder (relative to models/checkpoints or absolute), so a whole folder can be swept with only one model in memory at a time - Model Compare Sheet saves each run's image as a tile under output/model_compare/<run_name>/ and rebuilds a labeled contact sheet from disk every run, so the grid survives between independent prompts - Starting a sweep at index 0 archives stale tiles to a timestamped folder - Running past the last checkpoint raises a clear error instead of wrapping, so an over-queued batch never restarts and archives a finished sweep - Checkpoints load from the scanned absolute path, so folders outside the configured roots work as the tooltip promises - Extract the LoRA Folder Batch folder-scanning, natural-sort and regex filter helpers into kikotools/base/model_folders.py so both tools share one implementation
121 lines
5.2 KiB
Python
121 lines
5.2 KiB
Python
"""
|
|
KikoTools package initialization and node registry
|
|
Handles automatic discovery and registration of all ComfyAssets tools
|
|
"""
|
|
|
|
from .tools.batch_list_converter import (
|
|
ImageBatchToImageListNode,
|
|
ImageListToImageBatchNode,
|
|
LatentBatchToLatentListNode,
|
|
LatentListToLatentBatchNode,
|
|
)
|
|
from .tools.batch_prompts import BatchPromptsNode
|
|
from .tools.checkpoint_folder_iterator import CheckpointFolderIteratorNode
|
|
from .tools.display_any import DisplayAnyNode
|
|
from .tools.display_text import DisplayTextNode
|
|
from .tools.embedding_autocomplete import KikoEmbeddingAutocomplete
|
|
from .tools.empty_latent_batch import EmptyLatentBatchNode
|
|
from .tools.gemini_prompt import GeminiPromptNode
|
|
from .tools.image_scale_down_by import ImageScaleDownByNode
|
|
from .tools.image_to_multiple_of import ImageToMultipleOfNode
|
|
from .tools.kiko_film_grain import KikoFilmGrainNode
|
|
from .tools.kiko_purge_vram import KikoPurgeVRAM
|
|
from .tools.kiko_workflow_timer import KikoWorkflowTimerNode
|
|
from .tools.kiko_save_image import KikoSaveImageNode
|
|
from .tools.local_image_loader import LocalImageLoaderNode
|
|
from .tools.project_file_path import ProjectFilePathNode
|
|
from .tools.model_compare_sheet import ModelCompareSheetNode
|
|
from .tools.model_downloader import ModelDownloaderNode
|
|
from .tools.resolution_calculator import ResolutionCalculatorNode
|
|
from .tools.sampler_combo import SamplerComboCompactNode, SamplerComboNode
|
|
from .tools.seed_history import SeedHistoryNode
|
|
from .tools.text_input import TextInputNode
|
|
from .tools.width_height_selector import WidthHeightSelectorNode
|
|
from .tools.width_height_to_vec2 import WidthHeightToVec2Node
|
|
from .tools.xyz_helpers import (
|
|
FluxSamplerParamsNode,
|
|
LoRAFolderBatchNode,
|
|
PlotParametersNode,
|
|
SamplerSelectHelperNode,
|
|
SchedulerSelectHelperNode,
|
|
TextEncodeSamplerParamsNode,
|
|
)
|
|
|
|
# ComfyUI node registration mappings
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ImageBatchToImageList": ImageBatchToImageListNode,
|
|
"ImageListToImageBatch": ImageListToImageBatchNode,
|
|
"LatentBatchToLatentList": LatentBatchToLatentListNode,
|
|
"LatentListToLatentBatch": LatentListToLatentBatchNode,
|
|
"BatchPrompts": BatchPromptsNode,
|
|
"ResolutionCalculator": ResolutionCalculatorNode,
|
|
"WidthHeightSelector": WidthHeightSelectorNode,
|
|
"SeedHistory": SeedHistoryNode,
|
|
"SamplerCombo": SamplerComboNode,
|
|
"SamplerComboCompact": SamplerComboCompactNode,
|
|
"EmptyLatentBatch": EmptyLatentBatchNode,
|
|
"KikoSaveImage": KikoSaveImageNode,
|
|
"ImageToMultipleOf": ImageToMultipleOfNode,
|
|
"ImageScaleDownBy": ImageScaleDownByNode,
|
|
"GeminiPrompt": GeminiPromptNode,
|
|
"DisplayAny": DisplayAnyNode,
|
|
"DisplayText": DisplayTextNode,
|
|
"TextInput": TextInputNode,
|
|
"KikoFilmGrain": KikoFilmGrainNode,
|
|
"KikoPurgeVRAM": KikoPurgeVRAM,
|
|
"KikoLocalImageLoader": LocalImageLoaderNode,
|
|
"KikoModelDownloader": ModelDownloaderNode,
|
|
"SamplerSelectHelper": SamplerSelectHelperNode,
|
|
"SchedulerSelectHelper": SchedulerSelectHelperNode,
|
|
"TextEncodeSamplerParams": TextEncodeSamplerParamsNode,
|
|
"FluxSamplerParams": FluxSamplerParamsNode,
|
|
"PlotParameters+": PlotParametersNode,
|
|
"LoRAFolderBatch": LoRAFolderBatchNode,
|
|
"CheckpointFolderIterator": CheckpointFolderIteratorNode,
|
|
"ModelCompareSheet": ModelCompareSheetNode,
|
|
"WidthHeightToVec2": WidthHeightToVec2Node,
|
|
"KikoWorkflowTimer": KikoWorkflowTimerNode,
|
|
"KikoProjectFilePath": ProjectFilePathNode,
|
|
# Note: KikoEmbeddingAutocomplete is not registered as a node
|
|
# It's a settings-only feature accessed through ComfyUI settings menu
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ImageBatchToImageList": "Image Batch to Image List",
|
|
"ImageListToImageBatch": "Image List to Image Batch",
|
|
"LatentBatchToLatentList": "Latent Batch to Latent List",
|
|
"LatentListToLatentBatch": "Latent List to Latent Batch",
|
|
"BatchPrompts": "Batch Prompts",
|
|
"ResolutionCalculator": "Resolution Calculator",
|
|
"WidthHeightSelector": "Width Height Selector",
|
|
"SeedHistory": "Seed History",
|
|
"SamplerCombo": "Sampler Combo",
|
|
"SamplerComboCompact": "Sampler Combo (Compact)",
|
|
"EmptyLatentBatch": "Empty Latent Batch",
|
|
"KikoSaveImage": "Kiko Save Image",
|
|
"ImageToMultipleOf": "Image to Multiple of",
|
|
"ImageScaleDownBy": "Image Scale Down By",
|
|
"GeminiPrompt": "Gemini Prompt Engineer",
|
|
"DisplayAny": "Display Any",
|
|
"DisplayText": "Display Text",
|
|
"TextInput": "Text Input",
|
|
"KikoFilmGrain": "Film Grain",
|
|
"KikoPurgeVRAM": "Kiko Purge VRAM",
|
|
"KikoLocalImageLoader": "Local Image Loader",
|
|
"KikoModelDownloader": "Model Downloader 🌐",
|
|
"SamplerSelectHelper": "Sampler Select Helper",
|
|
"SchedulerSelectHelper": "Scheduler Select Helper",
|
|
"TextEncodeSamplerParams": "Text Encode for Sampler Params",
|
|
"FluxSamplerParams": "Flux Sampler Parameters",
|
|
"PlotParameters+": "Plot Parameters",
|
|
"LoRAFolderBatch": "LoRA Folder Batch",
|
|
"CheckpointFolderIterator": "Checkpoint Folder Iterator",
|
|
"ModelCompareSheet": "Model Compare Sheet",
|
|
"WidthHeightToVec2": "Width Height to VEC2",
|
|
"KikoWorkflowTimer": "Workflow Timer",
|
|
"KikoProjectFilePath": "Project File Path",
|
|
# KikoEmbeddingAutocomplete removed - settings only, not a node
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|