Eric Rollei
ea06cf9157
INT8 Instruct Fix (5 bugs): - blocks_to_swap no longer forced to 0 for INT8 in generate() - Added _load_int8_block_swap() method in loader with CB/SCB guard hooks - Fixed INT8 model size from 40GB to 80GB in memory budget - CB/SCB now fixed after .to(device) in _move_non_block_components_to_gpu() - Quant-specific gb_per_block in _calculate_optimal_config() VAE Decode Fix: - Fixed super() crash from closure cell corruption in cache clearing - Added isinstance(val, type) guard to prevent __class__ cell nuke - Cleaned up instance-level forward attrs after hook removal (4 files) New Features: - Experimental latent control nodes (Empty Latent, Noise Shaping, Generate with Latent) - Unified Generate V2 node (auto-detects NF4/INT8/BF16, block swap, memory mgmt) - HuggingFace upload script for v2 models (quantization/upload_v2_to_hf.py) README: - Updated to v1.3.0 with v2 HuggingFace model links - INT8 Instruct status changed to Working across all tables - Added Latent Control Nodes documentation section - Added v2 download links for base and Instruct models"
100 lines
3.8 KiB
Python
100 lines
3.8 KiB
Python
"""
|
|
HunyuanImage-3.0 ComfyUI Custom Nodes
|
|
Professional custom nodes for running Tencent's HunyuanImage-3.0 model in ComfyUI
|
|
|
|
Author: Eric Hiss (GitHub: EricRollei)
|
|
Contact: [eric@historic.camera, eric@rollei.us]
|
|
License: Dual License (Non-Commercial and Commercial Use)
|
|
Copyright (c) 2025-2026 Eric Hiss. All rights reserved.
|
|
|
|
Dual License:
|
|
1. Non-Commercial Use: This software is licensed under the terms of the
|
|
Creative Commons Attribution-NonCommercial 4.0 International License.
|
|
To view a copy of this license, visit http://creativecommons.org/licenses/by-nc/4.0/
|
|
|
|
2. Commercial Use: For commercial use, a separate license is required.
|
|
Please contact Eric Hiss at [eric@historic.camera, eric@rollei.us] for licensing options.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
|
|
INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
|
|
PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
|
|
|
Note: HunyuanImage-3.0 model is subject to Tencent's Apache 2.0 license.
|
|
"""
|
|
|
|
# Core working nodes - always loaded
|
|
from .hunyuan_quantized_nodes import NODE_CLASS_MAPPINGS as QUANTIZED_MAPPINGS
|
|
from .hunyuan_quantized_nodes import NODE_DISPLAY_NAME_MAPPINGS as QUANTIZED_DISPLAY_MAPPINGS
|
|
|
|
from .hunyuan_full_bf16_nodes import NODE_CLASS_MAPPINGS as FULL_MAPPINGS
|
|
from .hunyuan_full_bf16_nodes import NODE_DISPLAY_NAME_MAPPINGS as FULL_DISPLAY_MAPPINGS
|
|
|
|
from .hunyuan_api_nodes import NODE_CLASS_MAPPINGS as API_MAPPINGS
|
|
from .hunyuan_api_nodes import NODE_DISPLAY_NAME_MAPPINGS as API_DISPLAY_MAPPINGS
|
|
|
|
# V2 unified node and Instruct nodes - load conditionally
|
|
UNIFIED_V2_MAPPINGS = {}
|
|
UNIFIED_V2_DISPLAY_MAPPINGS = {}
|
|
INSTRUCT_MAPPINGS = {}
|
|
INSTRUCT_DISPLAY_MAPPINGS = {}
|
|
HIGHRES_MAPPINGS = {}
|
|
HIGHRES_DISPLAY_MAPPINGS = {}
|
|
MOE_TEST_MAPPINGS = {}
|
|
MOE_TEST_DISPLAY_MAPPINGS = {}
|
|
LATENT_MAPPINGS = {}
|
|
LATENT_DISPLAY_MAPPINGS = {}
|
|
|
|
try:
|
|
from .hunyuan_unified_v2 import NODE_CLASS_MAPPINGS as UNIFIED_V2_MAPPINGS
|
|
from .hunyuan_unified_v2 import NODE_DISPLAY_NAME_MAPPINGS as UNIFIED_V2_DISPLAY_MAPPINGS
|
|
except Exception as e:
|
|
print(f"[Eric_Hunyuan3] Warning: Could not load hunyuan_unified_v2: {e}")
|
|
|
|
try:
|
|
from .hunyuan_instruct_nodes import NODE_CLASS_MAPPINGS as INSTRUCT_MAPPINGS
|
|
from .hunyuan_instruct_nodes import NODE_DISPLAY_NAME_MAPPINGS as INSTRUCT_DISPLAY_MAPPINGS
|
|
except Exception as e:
|
|
print(f"[Eric_Hunyuan3] Warning: Could not load hunyuan_instruct_nodes: {e}")
|
|
|
|
try:
|
|
from .hunyuan_highres_nodes import NODE_CLASS_MAPPINGS as HIGHRES_MAPPINGS
|
|
from .hunyuan_highres_nodes import NODE_DISPLAY_NAME_MAPPINGS as HIGHRES_DISPLAY_MAPPINGS
|
|
except Exception as e:
|
|
print(f"[Eric_Hunyuan3] Warning: Could not load hunyuan_highres_nodes: {e}")
|
|
|
|
try:
|
|
from .hunyuan_moe_test_node import NODE_CLASS_MAPPINGS as MOE_TEST_MAPPINGS
|
|
from .hunyuan_moe_test_node import NODE_DISPLAY_NAME_MAPPINGS as MOE_TEST_DISPLAY_MAPPINGS
|
|
except Exception as e:
|
|
print(f"[Eric_Hunyuan3] Warning: Could not load hunyuan_moe_test_node: {e}")
|
|
|
|
try:
|
|
from .hunyuan_latent_nodes import NODE_CLASS_MAPPINGS as LATENT_MAPPINGS
|
|
from .hunyuan_latent_nodes import NODE_DISPLAY_NAME_MAPPINGS as LATENT_DISPLAY_MAPPINGS
|
|
except Exception as e:
|
|
print(f"[Eric_Hunyuan3] Warning: Could not load hunyuan_latent_nodes: {e}")
|
|
|
|
# Combine all mappings
|
|
NODE_CLASS_MAPPINGS = {
|
|
**QUANTIZED_MAPPINGS,
|
|
**FULL_MAPPINGS,
|
|
**API_MAPPINGS,
|
|
**UNIFIED_V2_MAPPINGS,
|
|
**INSTRUCT_MAPPINGS,
|
|
**HIGHRES_MAPPINGS,
|
|
**MOE_TEST_MAPPINGS,
|
|
**LATENT_MAPPINGS,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**QUANTIZED_DISPLAY_MAPPINGS,
|
|
**FULL_DISPLAY_MAPPINGS,
|
|
**API_DISPLAY_MAPPINGS,
|
|
**UNIFIED_V2_DISPLAY_MAPPINGS,
|
|
**INSTRUCT_DISPLAY_MAPPINGS,
|
|
**HIGHRES_DISPLAY_MAPPINGS,
|
|
**MOE_TEST_DISPLAY_MAPPINGS,
|
|
**LATENT_DISPLAY_MAPPINGS,
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |