Implemented spandrel

comfy/model_management.py

@@ -265,7 +265,10 @@ def get_torch_device_name(device):
 
 def module_size(module):
     module_mem = 0
-    sd = module.state_dict()
+    try:
+        sd = module.state_dict()
+    except:
+        sd = dict(module.__dict__.items())
     for k in sd:
         t = sd[k]
         module_mem += t.nelement() * t.element_size()

latent_preview.py

@@ -64,6 +64,8 @@ def get_previewer(device, latent_format):
 
         if method == LatentPreviewMethod.Auto:
             method = LatentPreviewMethod.Latent2RGB
+            if taesd_decoder_path:
+                method = LatentPreviewMethod.TAESD
 
         if method == LatentPreviewMethod.TAESD:
             if taesd_decoder_path:

requirements.txt

@@ -14,3 +14,4 @@ psutil
 #non essential dependencies:
 kornia>=0.7.1
 spandrel
+xformers