Add empty latent node for Sana
Also remove unused display name mappings
This commit is contained in:
+11
-7
@@ -1,5 +1,6 @@
|
|||||||
import torch
|
import torch
|
||||||
import folder_paths
|
import folder_paths
|
||||||
|
from nodes import EmptyLatentImage
|
||||||
|
|
||||||
from .conf import sana_conf, sana_res
|
from .conf import sana_conf, sana_res
|
||||||
from .loader import load_sana
|
from .loader import load_sana
|
||||||
@@ -36,6 +37,15 @@ class SanaCheckpointLoader:
|
|||||||
return (model,)
|
return (model,)
|
||||||
|
|
||||||
|
|
||||||
|
class EmptySanaLatentImage(EmptyLatentImage):
|
||||||
|
CATEGORY = "ExtraModels/Sana"
|
||||||
|
TITLE = "Empty Sana Latent Image"
|
||||||
|
|
||||||
|
def generate(self, width, height, batch_size=1):
|
||||||
|
latent = torch.zeros([batch_size, 32, height // 32, width // 32], device=self.device)
|
||||||
|
return ({"samples":latent}, )
|
||||||
|
|
||||||
|
|
||||||
class SanaResolutionSelect():
|
class SanaResolutionSelect():
|
||||||
@classmethod
|
@classmethod
|
||||||
def INPUT_TYPES(s):
|
def INPUT_TYPES(s):
|
||||||
@@ -138,11 +148,5 @@ NODE_CLASS_MAPPINGS = {
|
|||||||
"SanaResolutionSelect" : SanaResolutionSelect,
|
"SanaResolutionSelect" : SanaResolutionSelect,
|
||||||
"SanaTextEncode" : SanaTextEncode,
|
"SanaTextEncode" : SanaTextEncode,
|
||||||
"SanaResolutionCond" : SanaResolutionCond,
|
"SanaResolutionCond" : SanaResolutionCond,
|
||||||
}
|
"EmptySanaLatentImage": EmptySanaLatentImage,
|
||||||
|
|
||||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
||||||
"Sana Checkpoint Loader": "SanaCheckpointLoader",
|
|
||||||
"Sana Resolution Select": "SanaResolutionSelect",
|
|
||||||
"Sana Text Encoder": "SanaTextEncode",
|
|
||||||
"Sana Resolution Cond": "SanaResolutionCond",
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user