33 lines
913 B
Python
33 lines
913 B
Python
from .ltx_director import LTXDirector
|
|
from .ltx_director_guide import LTXDirectorGuide
|
|
from comfy_api.latest import ComfyExtension, io
|
|
from typing_extensions import override
|
|
from .latent_slice import CleanLatentSlice
|
|
|
|
class PromptRelay(ComfyExtension):
|
|
@override
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
LTXDirector,
|
|
LTXDirectorGuide
|
|
]
|
|
|
|
async def comfy_entrypoint() -> PromptRelay:
|
|
return PromptRelay()
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LTXDirectorCS": LTXDirector,
|
|
"LTXDirectorGuideCS": LTXDirectorGuide,
|
|
"CleanLatentSliceCS": CleanLatentSlice,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LTXDirectorCS": "LTX Director CS",
|
|
"LTXDirectorGuideCS": "LTX Director Guide CS",
|
|
"CleanLatentSliceCS": "Clean Latent Slice CS",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./js"
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']
|