32 lines
858 B
Python
32 lines
858 B
Python
import os
|
|
|
|
from comfy_api.latest import ComfyExtension, io
|
|
|
|
from .nodes.gen_image_node import GenerateImage
|
|
from .nodes.gen_speech_node import GenerateSpeech
|
|
from .nodes.gen_text_node import GenerateTextAdvanced
|
|
from .nodes.gen_video_from_text_node import GenerateVideoFromText
|
|
from .nodes.i2i_enhance_upscale import I2IEnhanceUpscale
|
|
from .nodes.test_node import DCTestNode
|
|
|
|
|
|
class VeniceExtension(ComfyExtension):
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
DCTestNode,
|
|
GenerateImage,
|
|
GenerateSpeech,
|
|
GenerateTextAdvanced,
|
|
GenerateVideoFromText,
|
|
I2IEnhanceUpscale,
|
|
]
|
|
|
|
|
|
async def comfy_entrypoint() -> VeniceExtension:
|
|
return VeniceExtension()
|
|
|
|
|
|
WEB_DIRECTORY = os.path.join(os.path.dirname(__file__), "js")
|
|
|
|
__all__ = ["WEB_DIRECTORY"]
|