Fix VAE Selector input types

This commit is contained in:
Thomas Ward
2024-05-11 19:01:59 -04:00
parent d310013ff3
commit 52b8b02bee
+8
View File
@@ -86,6 +86,14 @@ class HelperNodes_VAESelector(BaseNode):
sd["vae_scale"] = torch.tensor(0.13025)
return sd
@classmethod
def INPUT_TYPES(cls) -> dict:
return {
"required": {
"vae_name": (cls.vae_list(),)
}
}
CATEGORY = f"{MODULE_CATEGORY}"
RETURN_TYPES = ("VAE",)