Implement context management system: - base_context.py: foundational context node - context.py: standard context handling - context_passthrough.py: pass-through node - even_larger_context.py: maximum context capacity - large_context_flux.py: Flux-specific large context - large_context_sdxl.py: SDXL-specific large context Add markdown documentation for all context nodes
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from .nodes.context.base_context import TT_BaseContext
|
|
from .nodes.context.context import TT_Context
|
|
from .nodes.context.context_passthrough import TT_ContextPassthrough
|
|
from .nodes.context.even_larger_context import TT_EvenLargerContext
|
|
from .nodes.context.large_context_flux import TT_LargeContextFlux
|
|
from .nodes.context.large_context_sdxl import TT_LargeContextSdxl
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
# Context
|
|
"TT_BaseContext": TT_BaseContext,
|
|
"TT_Context": TT_Context,
|
|
"TT_LargeContextFlux": TT_LargeContextFlux,
|
|
"TT_LargeContextSdxl": TT_LargeContextSdxl,
|
|
"TT_EvenLargerContext": TT_EvenLargerContext,
|
|
# Context Passthrough
|
|
"TT_ContextPassthrough": TT_ContextPassthrough,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
# Context
|
|
"TT_BaseContext": "TT Base Context",
|
|
"TT_Context": "TT Context",
|
|
"TT_LargeContextFlux": "TT FLUX Large Context",
|
|
"TT_LargeContextSdxl": "TT SDXL Large Context",
|
|
"TT_EvenLargerContext": "TT Even Larger Context",
|
|
# Context Passthrough
|
|
"TT_ContextPassthrough": "TT Context Passthrough",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']
|