Sana loader logic

2024-12-11 18:55:17 +01:00
parent 06a936813f
commit 5505ab4f40
4 changed files with 142 additions and 319 deletions
@@ -2,9 +2,11 @@ import folder_paths
 import comfy.utils

 from .PixArt.loader import load_pixart_state_dict
+from .Sana.loader import load_sana_state_dict

 loaders = {
    "PixArt": load_pixart_state_dict,
+    "Sana": load_sana_state_dict,
 }

 class EXMUnetLoader: