Enforce ComfyExtension types (#1718)

* Enforce extension types

* nit
This commit is contained in:
Chenlei Hu
2024-11-27 12:35:18 -08:00
committed by GitHub
parent 39eaa2e850
commit c4d3c672ad
7 changed files with 27 additions and 4 deletions

View File

@@ -81,6 +81,7 @@ app.registerExtension({
name: 'Comfy.AudioWidget',
async beforeRegisterNodeDef(nodeType, nodeData) {
if (
// @ts-expect-error ComfyNode
['LoadAudio', 'SaveAudio', 'PreviewAudio'].includes(nodeType.comfyClass)
) {
nodeData.input.required.audioUI = ['AUDIO_UI']