nodes/WJLUOXIAO-XB_ToolBox
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

完善int8节点

Browse Source
This commit is contained in:
AI_Architect
2026-07-13 02:27:05 +08:00
parent f6e83e006b
commit c7fb69474f
4 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+3 -3
View File
@@ -87,7 +87,7 @@ try:
from .nodes_comic import XB_ComicPromptParser, XB_ComicTextRenderer, XB_AutoBubbleTextRenderer
from .nodes_super import XB_BerniniPromptRouter, XB_K2StyleRouter
from .nodes_int8_rocm import (
UNetLoaderINTW8A8ROCm, INT8GroupedLoraROCm, INT8LoraROCm, INT8ModelSaveROCm, PreLoraLoaderROCm
UNetLoaderINTROCm, INT8GroupedLoraROCm, INT8LoraROCm, INT8ModelSaveROCm, PreLoraLoaderROCm
)
from .nodes_int8_clip_rocm import (
CLIPLoaderINT8ROCm, DualCLIPLoaderINT8ROCm, INT8CLIPSaveROCm
@@ -191,7 +191,7 @@ try:
"XB_BerniniPromptRouter": XB_BerniniPromptRouter,
"XB_K2StyleRouter": XB_K2StyleRouter,
# ── INT8 ROCm 量化节点 ──
"XB_UNetLoaderINTW8A8ROCm": UNetLoaderINTW8A8ROCm,
"XB_UNetLoaderINTROCm": UNetLoaderINTROCm,
"XB_INT8LoraROCm": INT8LoraROCm,
"XB_INT8GroupedLoraROCm": INT8GroupedLoraROCm,
"XB_INT8ModelSaveROCm": INT8ModelSaveROCm,
@@ -286,7 +286,7 @@ try:
"XB_BerniniPromptRouter": "XB-BOX - 🧠 Bernini 提示词路由（超级节点）",
"XB_K2StyleRouter": "XB-BOX - 🎨 K2 风格切换路由（超级节点）",
# ── INT8 ROCm 量化节点 ──
"XB_UNetLoaderINTW8A8ROCm": "XB-BOX - 🧊 INT8 UNet 加载器 (W8A8 ROCm)",
"XB_UNetLoaderINTROCm": "XB-BOX - 🧊 INT8 UNet 加载器 (ROCm)",
"XB_INT8LoraROCm": "XB-BOX - 🧊 INT8 单LoRA (ROCm)",
"XB_INT8GroupedLoraROCm": "XB-BOX - 🧊 INT8 组合 LoRA (ROCm)",
"XB_INT8ModelSaveROCm": "XB-BOX - 🧊 INT8 模型保存 (ROCm)",
locales/zh/nodeDefs.json
+2 -2
View File
@@ -3786,8 +3786,8 @@
}
}
},
"XB_UNetLoaderINTW8A8ROCm": {
"display_name": "XB-BOX - 🧊 INT8 UNet 加载器 (W8A8 ROCm)",
"XB_UNetLoaderINTROCm": {
"display_name": "XB-BOX - 🧊 INT8 UNet 加载器 (ROCm)",
"inputs": {
"unet_name": {
"name": "UNet模型"
nodes_int8_clip_rocm.py
+2 -2
View File
@@ -135,7 +135,7 @@ class CLIPLoaderINT8ROCm:
RETURN_TYPES = ("CLIP",)
FUNCTION = "load_clip"
CATEGORY = "loaders"
TITLE = "Load CLIP INT8 ROCm (W8A8)"
TITLE = "Load CLIP INT8 ROCm"
DESCRIPTION = "XB INT8 ROCm: 加载INT8文本编码器"
def load_clip(self, clip_name, type="wan", weight_dtype="default",
@@ -170,7 +170,7 @@ class DualCLIPLoaderINT8ROCm(CLIPLoaderINT8ROCm):
},
}
TITLE = "Load Dual CLIP INT8 ROCm (W8A8)"
TITLE = "Load Dual CLIP INT8 ROCm"
DESCRIPTION = "XB INT8 ROCm: 加载双INT8文本编码器"
def load_clip(self, clip_name1, clip_name2, type="flux", weight_dtype="default",
nodes_int8_rocm.py
+6 -6
View File
@@ -1,10 +1,10 @@
"""
XB-ToolBox INT8 ROCm 节点
========================
基于 ComfyUI-INT8-Fast-ROCM 适配，提供 INT8 W8A8 量化推理支持。
基于 ComfyUI-INT8-Fast-ROCM 适配，提供 INT8 量化推理支持。
节点列表：
- XB_UNetLoaderINTW8A8ROCm : INT8 UNet 模型加载器
- XB_UNetLoaderINTROCm : INT8 UNet 模型加载器
- XB_INT8GroupedLoraROCm : INT8 组合 LoRA
- XB_INT8ModelSaveROCm : INT8 模型保存
- XB_INT8PreLoraLoaderROCm : INT8 预加载 LoRA
@@ -1169,20 +1169,20 @@ class PreLoraLoaderROCm:
# =============================================================================
# UNetLoaderINTW8A8ROCm 节点
# UNetLoaderINTROCm 节点
# =============================================================================
def _load_int8_unet_cached_patcher_rocm(unet_name, weight_dtype, model_type,
on_the_fly_quantization, enable_convrot,
lora_mode, pre_lora=None, disable_dynamic=False):
return UNetLoaderINTW8A8ROCm().load_unet(
return UNetLoaderINTROCm().load_unet(
unet_name, weight_dtype, model_type, on_the_fly_quantization,
enable_convrot=enable_convrot, lora_mode=lora_mode,
pre_lora=pre_lora, disable_dynamic=disable_dynamic,
)[0]
class UNetLoaderINTW8A8ROCm:
class UNetLoaderINTROCm:
@classmethod
def INPUT_TYPES(s):
return {
@@ -1201,7 +1201,7 @@ class UNetLoaderINTW8A8ROCm:
RETURN_TYPES = ("MODEL",)
FUNCTION = "load_unet"
CATEGORY = "loaders"
DESCRIPTION = "XB INT8 ROCm: 加载INT8量化扩散模型 (W8A8)"
DESCRIPTION = "XB INT8 ROCm: 加载INT8量化扩散模型"
def load_unet(self, unet_name, weight_dtype, model_type, on_the_fly_quantization,
enable_convrot=False, lora_mode="None", pre_lora=None, disable_dynamic=False):