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
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
"""
|
|
Core node implementations for ROCM Ninodes.
|
|
|
|
This module contains all ComfyUI node classes organized by functionality:
|
|
- VAE nodes (decode operations)
|
|
- Sampler nodes (generation/sampling)
|
|
- Checkpoint loader nodes
|
|
- UNet/Diffusion Model loader nodes
|
|
- LoRA loader nodes
|
|
- Monitor/utility nodes
|
|
"""
|
|
|
|
from .vae import (
|
|
ROCMOptimizedVAEDecode,
|
|
ROCMOptimizedVAEDecodeTiled,
|
|
ROCMVAEPerformanceMonitor,
|
|
)
|
|
from .sampler import (
|
|
ROCMOptimizedKSampler,
|
|
ROCMOptimizedKSamplerAdvanced,
|
|
ROCMSamplerPerformanceMonitor,
|
|
ROCMSamplerCustomAdvanced,
|
|
ROCMSamplerCustomAdvancedBenchmark,
|
|
)
|
|
from .checkpoint import ROCMOptimizedCheckpointLoader
|
|
from .unet_loader import ROCmDiffusionLoader
|
|
from .lora import ROCMLoRALoader
|
|
from .monitors import ROCMFluxBenchmark, ROCMMemoryOptimizer
|
|
from .textgen_ltx2 import ROCmTextGenerateLTX2Prompt
|
|
|
|
__all__ = [
|
|
# VAE nodes
|
|
'ROCMOptimizedVAEDecode',
|
|
'ROCMOptimizedVAEDecodeTiled',
|
|
'ROCMVAEPerformanceMonitor',
|
|
# Sampler nodes
|
|
'ROCMOptimizedKSampler',
|
|
'ROCMOptimizedKSamplerAdvanced',
|
|
'ROCMSamplerPerformanceMonitor',
|
|
'ROCMSamplerCustomAdvanced',
|
|
'ROCMSamplerCustomAdvancedBenchmark',
|
|
# Loader nodes
|
|
'ROCMOptimizedCheckpointLoader',
|
|
'ROCmDiffusionLoader',
|
|
'ROCMLoRALoader',
|
|
# Monitor nodes
|
|
'ROCMFluxBenchmark',
|
|
'ROCMMemoryOptimizer',
|
|
'ROCmTextGenerateLTX2Prompt',
|
|
]
|
|
|