nodes/space-nuko-nui-suite
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Don't encode conditioning internally

Browse Source
This commit is contained in:
space-nuko
2023-06-04 09:53:31 -05:00
parent abc38b1cfd
commit 19e7881c6b
1 changed files with 12 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+12 -14
View File
@@ -12,14 +12,13 @@ NODE_FILE = os.path.abspath(__file__)
NUI_SUITE_ROOT = os.path.dirname(NODE_FILE) NUI_SUITE_ROOT = os.path.dirname(NODE_FILE)
class DynamicPromptsTextEncode: class DynamicPromptsTextGen:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": {"text": ("STRING", {"multiline": True}), return {"required": {"text": ("STRING", {"multiline": True}),
"clip": ("CLIP", ),
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}), "seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}),
}} }}
RETURN_TYPES = ("CONDITIONING", "STRING",) RETURN_TYPES = ("STRING",)
FUNCTION = "encode" FUNCTION = "encode"
CATEGORY = "conditioning" CATEGORY = "conditioning"
@@ -33,7 +32,7 @@ class DynamicPromptsTextEncode:
wildcard_wrap="__" wildcard_wrap="__"
) )
def encode(self, text, clip, seed): def encode(self, text, seed):
prompt_generator = RandomPromptGenerator( prompt_generator = RandomPromptGenerator(
self._wildcard_manager, self._wildcard_manager,
seed=seed, seed=seed,
@@ -45,17 +44,16 @@ class DynamicPromptsTextEncode:
all_prompts = prompt_generator.generate(text, 1) or [""] all_prompts = prompt_generator.generate(text, 1) or [""]
prompt = all_prompts[0] prompt = all_prompts[0]
return ([[clip.encode(prompt), {}]], prompt, ) return (prompt, )
class FeelingLuckyTextEncode: class FeelingLuckyTextGen:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": {"text": ("STRING", {"multiline": True}), return {"required": {"text": ("STRING", {"multiline": True}),
"clip": ("CLIP", ),
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}), "seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}),
}} }}
RETURN_TYPES = ("CONDITIONING","STRING",) RETURN_TYPES = ("STRING",)
FUNCTION = "encode" FUNCTION = "encode"
CATEGORY = "conditioning" CATEGORY = "conditioning"
@@ -69,7 +67,7 @@ class FeelingLuckyTextEncode:
wildcard_wrap="__" wildcard_wrap="__"
) )
def encode(self, text, clip, seed): def encode(self, text, seed):
inner_generator = RandomPromptGenerator( inner_generator = RandomPromptGenerator(
self._wildcard_manager, self._wildcard_manager,
seed=seed, seed=seed,
@@ -82,7 +80,7 @@ class FeelingLuckyTextEncode:
all_prompts = prompt_generator.generate(text, 1) or [""] all_prompts = prompt_generator.generate(text, 1) or [""]
prompt = all_prompts[0] prompt = all_prompts[0]
return ([[clip.encode(prompt), {}]], prompt, ) return (prompt, )
class OutputString: class OutputString:
@@ -106,13 +104,13 @@ class OutputString:
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"Nui.DynamicPromptsTextEncode": DynamicPromptsTextEncode, "Nui.DynamicPromptsTextGen": DynamicPromptsTextGen,
"Nui.FeelingLuckyTextEncode": FeelingLuckyTextEncode, "Nui.FeelingLuckyTextGen": FeelingLuckyTextGen,
"Nui.OutputString": OutputString, "Nui.OutputString": OutputString,
} }
NODE_DISPLAY_NAME_MAPPINGS = { NODE_DISPLAY_NAME_MAPPINGS = {
"Nui.DynamicPromptsTextEncode": "Dynamic Prompts Text Encode", "Nui.DynamicPromptsTextEncode": "Dynamic Prompts Text Generator",
"Nui.FeelingLuckyTextEncode": "Feeling Lucky Text Encode", "Nui.FeelingLuckyTextEncode": "Feeling Lucky Text Generator",
"Nui.OutputString": "Output String", "Nui.OutputString": "Output String",
} }