iGavroche
20626b4fd4
- ROCMSamplerCustomAdvanced: V3 ComfyNode drop-in for stock SamplerCustomAdvanced with architecture detection, ROCm backend tuning, memory defrag for high-memory models (LTX 128ch), enhanced per-step callback with ETA - ROCMSamplerCustomAdvancedBenchmark: A/B comparison node for stock vs ROCm - docs/SAMPLER_CUSTOM_ADVANCED.md: full documentation for agents and users - Category: ROCm Ninodes/Sampling - Bump version to 2.2.0 across all files
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
"""
|
|
Node registry for ROCM Ninodes.
|
|
|
|
This module imports all node classes and creates the NODE_CLASS_MAPPINGS
|
|
and NODE_DISPLAY_NAME_MAPPINGS required by ComfyUI.
|
|
"""
|
|
|
|
# Import all node classes from core modules
|
|
from .core.vae import (
|
|
ROCMOptimizedVAEDecode,
|
|
ROCMOptimizedVAEDecodeTiled,
|
|
ROCMVAEPerformanceMonitor,
|
|
)
|
|
|
|
from .core.sampler import (
|
|
ROCMOptimizedKSampler,
|
|
ROCMOptimizedKSamplerAdvanced,
|
|
ROCMSamplerPerformanceMonitor,
|
|
ROCMSamplerCustomAdvanced,
|
|
ROCMSamplerCustomAdvancedBenchmark,
|
|
)
|
|
|
|
from .core.checkpoint import (
|
|
ROCMOptimizedCheckpointLoader,
|
|
)
|
|
|
|
from .core.unet_loader import (
|
|
ROCmDiffusionLoader,
|
|
)
|
|
|
|
from .core.gguf_loader import (
|
|
ROCmGGUFLoader,
|
|
)
|
|
|
|
from .core.lora import (
|
|
ROCMLoRALoader,
|
|
)
|
|
|
|
from .core.monitors import (
|
|
ROCMFluxBenchmark,
|
|
ROCMMemoryOptimizer,
|
|
)
|
|
from .core.textgen_ltx2 import ROCmTextGenerateLTX2Prompt
|
|
|
|
# Define node class mappings for ComfyUI
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ROCMOptimizedCheckpointLoader": ROCMOptimizedCheckpointLoader,
|
|
"ROCmDiffusionLoader": ROCmDiffusionLoader,
|
|
"ROCmGGUFLoader": ROCmGGUFLoader,
|
|
"ROCMOptimizedVAEDecode": ROCMOptimizedVAEDecode,
|
|
"ROCMOptimizedVAEDecodeTiled": ROCMOptimizedVAEDecodeTiled,
|
|
"ROCMVAEPerformanceMonitor": ROCMVAEPerformanceMonitor,
|
|
"ROCMOptimizedKSampler": ROCMOptimizedKSampler,
|
|
"ROCMOptimizedKSamplerAdvanced": ROCMOptimizedKSamplerAdvanced,
|
|
"ROCMSamplerCustomAdvanced": ROCMSamplerCustomAdvanced,
|
|
"ROCMSamplerPerformanceMonitor": ROCMSamplerPerformanceMonitor,
|
|
"ROCMSamplerCustomAdvancedBenchmark": ROCMSamplerCustomAdvancedBenchmark,
|
|
"ROCMFluxBenchmark": ROCMFluxBenchmark,
|
|
"ROCMMemoryOptimizer": ROCMMemoryOptimizer,
|
|
"ROCMLoRALoader": ROCMLoRALoader,
|
|
"ROCmTextGenerateLTX2Prompt": ROCmTextGenerateLTX2Prompt,
|
|
}
|
|
|
|
# Define display name mappings for ComfyUI
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ROCMOptimizedCheckpointLoader": "ROCm Checkpoint Loader",
|
|
"ROCmDiffusionLoader": "ROCm Diffusion Loader",
|
|
"ROCmGGUFLoader": "ROCm GGUF Loader",
|
|
"ROCMOptimizedVAEDecode": "ROCm VAE Decode",
|
|
"ROCMOptimizedVAEDecodeTiled": "ROCm VAE Decode Tiled",
|
|
"ROCMVAEPerformanceMonitor": "ROCm VAE Performance Monitor",
|
|
"ROCMOptimizedKSampler": "ROCm KSampler",
|
|
"ROCMOptimizedKSamplerAdvanced": "ROCm KSampler Advanced",
|
|
"ROCMSamplerCustomAdvanced": "ROCm SamplerCustomAdvanced",
|
|
"ROCMSamplerPerformanceMonitor": "ROCm Sampler Performance Monitor",
|
|
"ROCMSamplerCustomAdvancedBenchmark": "ROCm SamplerCustomAdvanced Benchmark",
|
|
"ROCMFluxBenchmark": "ROCm Flux Benchmark",
|
|
"ROCMMemoryOptimizer": "ROCm Memory Optimizer",
|
|
"ROCMLoRALoader": "ROCm LoRA Loader",
|
|
"ROCmTextGenerateLTX2Prompt": "ROCm Text Generate LTX2 Prompt",
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
|
|