23 lines
617 B
Python
23 lines
617 B
Python
from comfy_api.latest import ComfyExtension, io
|
|
from typing_extensions import override
|
|
|
|
from .yue_node import YUE_SM_Model,YUE_SM_Vae, YUE_SM_Sampler,YUE_SM_Clip,YUE_SM_Cond
|
|
|
|
WEB_DIRECTORY = "./js"
|
|
|
|
class YUE_SM_Extension(ComfyExtension):
|
|
@override
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
YUE_SM_Model,
|
|
YUE_SM_Vae,
|
|
YUE_SM_Sampler,
|
|
YUE_SM_Clip,
|
|
YUE_SM_Cond,
|
|
]
|
|
|
|
async def comfy_entrypoint() -> YUE_SM_Extension: # ComfyUI calls this to load your extension and its nodes.
|
|
return YUE_SM_Extension()
|
|
|
|
|