1

VAELoaderDecode.py

@@ -0,0 +1,75 @@
+import os
+import comfy.sd
+from nodes import *
+import folder_paths
+
+
+# VAEDecode
+# VAELoader
+# VAELoaderDecode
+
+#wd = os.getcwd()
+#print("working directory is ", wd)
+#
+#filePath = __file__
+#print("This script file path is ", filePath)
+#
+#absFilePath = os.path.abspath(__file__)
+#print("This script absolute path is ", absFilePath)
+#
+#realFilePath = os.path.realpath(__file__)
+#print("This script real path is ", realFilePath)
+#
+#path, filename = os.path.split(absFilePath)
+#print("Script file path is {}, filename is {}".format(path, filename))
+        
+class VAELoaderDecode:
+
+    def __init__(self, device="cpu"):
+        self.device = device
+
+    #@classmethod
+    #def INPUT_TYPES(s):
+    #    return {"required": { "samples": ("LATENT", ), "vae": ("VAE", )}}
+    #    
+    #@classmethod
+    #def INPUT_TYPES(s):
+    #    return {"required": { "vae_name": (filter_files_extensions(recursive_search(s.vae_dir), supported_pt_extensions), )}}
+        
+    @classmethod
+    def INPUT_TYPES(s):
+
+        return {
+            "required": { 
+                "samples": ("LATENT", ),
+                "vae_name": (folder_paths.get_filename_list("vae"), )
+            }
+        }
+        
+    RETURN_TYPES = ("IMAGE",)
+    
+    FUNCTION = "test"
+
+    CATEGORY = "latent"
+
+    #TODO: scale factor?
+    #def load_vae(self, vae_name):
+    #    vae_path = os.path.join(self.vae_dir, vae_name)
+    #    vae = comfy.sd.VAE(ckpt_path=vae_path)
+    #    return (vae,)
+    #    
+    #def decode(self, vae, samples):
+    #    return (vae.decode(samples["samples"]), )        
+        
+    def test(self, vae_name, samples):
+
+        t=folder_paths.get_filename_list("vae")
+        print(f"VAELoaderDecode : {t}")
+        vae_path = folder_paths.get_full_path("vae", vae_name)
+        print(f"VAELoaderDecode : {vae_path}")
+        vae = comfy.sd.VAE(ckpt_path=vae_path)
+        return (vae.decode(samples["samples"]), )
+        
+#NODE_CLASS_MAPPINGS = {
+#    "VAELoaderDecode": VAELoaderDecode,
+#}