41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from .ltx_director import LTXDirector
|
|
from .ltx_director_guide import LTXDirectorGuide, LTXDirectorCropGuides
|
|
from .msr_character import MSRCharacter, MSRCharacterSet
|
|
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 [
|
|
MSRCharacter,
|
|
MSRCharacterSet,
|
|
LTXDirector,
|
|
]
|
|
|
|
async def comfy_entrypoint() -> PromptRelay:
|
|
return PromptRelay()
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LTXDirectorCS": LTXDirector,
|
|
"MSRCharacterCS": MSRCharacter,
|
|
"MSRCharacterSetCS": MSRCharacterSet,
|
|
"LTXDirectorGuideCS": LTXDirectorGuide,
|
|
"LTXDirectorCropGuidesCS": LTXDirectorCropGuides,
|
|
"CleanLatentSliceCS": CleanLatentSlice,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LTXDirectorCS": "LTX Director CS",
|
|
"MSRCharacterCS": "MSR Character CS",
|
|
"MSRCharacterSetCS": "MSR Character Set CS",
|
|
"LTXDirectorGuideCS": "LTX Director Guide CS",
|
|
"LTXDirectorCropGuidesCS": "LTX Director Crop Guides CS",
|
|
"CleanLatentSliceCS": "Clean Latent Slice CS",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./js"
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']
|