XinPeng76
d0b104935c
- 新增TJSCustom节点，镜像ComfyUI原生SamplerCustom接口 - 支持model/add_noise/noise_seed/cfg/positive/negative/sampler/sigmas/latent_image输入 - 额外增加early_exit_gamma参数控制TJS早退 - 使用单次sample_custom调用，通过callback捕获x0和xt - 更新__init__.py、nodes.py和README.md
38 lines
985 B
Python
38 lines
985 B
Python
"""Compatibility wrapper for older imports.
|
|
|
|
ComfyUI loads this custom node through __init__.py, which imports tjs_nodes.
|
|
This file re-exports the same mappings in case a local workflow imports
|
|
ComfyUI_TJS.nodes directly.
|
|
"""
|
|
|
|
try:
|
|
from .tjs_nodes import (
|
|
NODE_CLASS_MAPPINGS as _SAMPLER_CLASS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as _SAMPLER_DISPLAY,
|
|
TJSAdvancedSampler,
|
|
TJSCustom,
|
|
TJSCustomAdvanced,
|
|
TJSSampler,
|
|
)
|
|
except ImportError:
|
|
from tjs_nodes import ( # type: ignore
|
|
NODE_CLASS_MAPPINGS as _SAMPLER_CLASS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as _SAMPLER_DISPLAY,
|
|
TJSAdvancedSampler,
|
|
TJSCustom,
|
|
TJSCustomAdvanced,
|
|
TJSSampler,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {**_SAMPLER_CLASS}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {**_SAMPLER_DISPLAY}
|
|
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
"TJSAdvancedSampler",
|
|
"TJSCustom",
|
|
"TJSCustomAdvanced",
|
|
"TJSSampler",
|
|
]
|