|
|
|
|
|
|
|
|
|
|
|
import torch |
|
import numpy as np |
|
from PIL import Image, ImageEnhance |
|
import os |
|
import sys |
|
import folder_paths |
|
from nodes import MAX_RESOLUTION, ControlNetApply |
|
from ..categories import icons |
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "comfy")) |
|
|
|
|
|
|
|
|
|
class CR_PromptMixPresets: |
|
def __init__(self): |
|
pass |
|
|
|
@classmethod |
|
def INPUT_TYPES(s): |
|
return { |
|
"required":{ |
|
}, |
|
"optional":{ |
|
"prompt_positive": ("STRING", {"multiline": True, "default": "prompt_pos"}), |
|
"prompt_negative": ("STRING", {"multiline": True, "default": "prompt_neg"}), |
|
"style_positive": ("STRING", {"multiline": True, "default": "style_pos"}), |
|
"style_negative": ("STRING", {"multiline": True, "default": "style_neg"}), |
|
"preset": (["default with no style text", "default with style text", |
|
"style boost 1", "style boost 2", "style text to refiner"],), |
|
}, |
|
} |
|
|
|
RETURN_TYPES = ("STRING", "STRING", "STRING", "STRING", "STRING", "STRING", "STRING", ) |
|
RETURN_NAMES = ("pos_g", "pos_l", "pos_r", "neg_g", "neg_l", "neg_r", "show_help", ) |
|
FUNCTION = "mixer" |
|
CATEGORY = icons.get("Comfyroll/SDXL") |
|
|
|
def mixer(self, prompt_positive, prompt_negative, style_positive, style_negative, preset): |
|
if preset == "default with no style text": |
|
pos_g = prompt_positive |
|
pos_l = prompt_positive |
|
pos_r = prompt_positive |
|
neg_g = prompt_negative |
|
neg_l = prompt_negative |
|
neg_r = prompt_negative |
|
elif preset == "default with style text": |
|
pos_g = prompt_positive + style_positive |
|
pos_l = prompt_positive + style_positive |
|
pos_r = prompt_positive + style_positive |
|
neg_g = prompt_negative + style_negative |
|
neg_l = prompt_negative + style_negative |
|
neg_r = prompt_negative + style_negative |
|
elif preset == "style boost 1": |
|
pos_g = prompt_positive |
|
pos_l = style_positive |
|
pos_r = prompt_positive |
|
neg_g = prompt_negative |
|
neg_l = style_negative |
|
neg_r = prompt_negative |
|
elif preset == "style boost 2": |
|
pos_g = style_positive |
|
pos_l = prompt_positive |
|
pos_r = style_positive |
|
neg_g = style_negative |
|
neg_l = prompt_negative |
|
neg_r = style_negative |
|
elif preset == "style text to refiner": |
|
pos_g = prompt_positive |
|
pos_l = prompt_positive |
|
pos_r = style_positive |
|
neg_g = prompt_negative |
|
neg_l = prompt_negative |
|
neg_r = style_negative |
|
show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/SDXL-Nodes#cr-sdxl-prompt-mix-presets" |
|
return (pos_g, pos_l, pos_r, neg_g, neg_l, neg_r, show_help, ) |
|
|
|
|
|
class CR_SDXLStyleText: |
|
@classmethod |
|
def INPUT_TYPES(s): |
|
return {"required": { |
|
"positive_style": ("STRING", {"default": "POS_STYLE", "multiline": True}), |
|
"negative_style": ("STRING", {"default": "NEG_STYLE", "multiline": True}), |
|
}, |
|
} |
|
|
|
RETURN_TYPES = ("STRING", "STRING", "STRING", ) |
|
RETURN_NAMES = ("positive_prompt_text_l", "negative_prompt_text_l" , "show_help", ) |
|
FUNCTION = "get_value" |
|
CATEGORY = icons.get("Comfyroll/SDXL") |
|
|
|
def get_value(self, positive_style, negative_style): |
|
show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/SDXL-Nodes#cr-sdxl-style-text" |
|
return (positive_style, negative_style, show_help, ) |
|
|
|
|
|
class CR_SDXLBasePromptEncoder: |
|
@classmethod |
|
def INPUT_TYPES(s): |
|
return {"required": { |
|
"base_clip": ("CLIP", ), |
|
"pos_g": ("STRING", {"multiline": True, "default": "POS_G"}), |
|
"pos_l": ("STRING", {"multiline": True, "default": "POS_L"}), |
|
"neg_g": ("STRING", {"multiline": True, "default": "NEG_G"}), |
|
"neg_l": ("STRING", {"multiline": True, "default": "NEG_L"}), |
|
"preset": (["preset A", "preset B", "preset C"],), |
|
"base_width": ("INT", {"default": 4096.0, "min": 0, "max": MAX_RESOLUTION, "step": 64}), |
|
"base_height": ("INT", {"default": 4096.0, "min": 0, "max": MAX_RESOLUTION, "step": 64}), |
|
"crop_w": ("INT", {"default": 0, "min": 0, "max": MAX_RESOLUTION, "step": 64}), |
|
"crop_h": ("INT", {"default": 0, "min": 0, "max": MAX_RESOLUTION, "step": 64}), |
|
"target_width": ("INT", {"default": 4096.0, "min": 0, "max": MAX_RESOLUTION, "step": 64}), |
|
"target_height": ("INT", {"default": 4096.0, "min": 0, "max": MAX_RESOLUTION, "step": 64}), |
|
}, |
|
} |
|
|
|
RETURN_TYPES = ("CONDITIONING", "CONDITIONING", "STRING", ) |
|
RETURN_NAMES = ("base_positive", "base_negative", "show_help", ) |
|
FUNCTION = "encode" |
|
CATEGORY = icons.get("Comfyroll/SDXL") |
|
|
|
def encode(self, base_clip, pos_g, pos_l, neg_g, neg_l, base_width, base_height, crop_w, crop_h, target_width, target_height, preset,): |
|
empty = base_clip.tokenize("") |
|
|
|
|
|
tokens1 = base_clip.tokenize(pos_g) |
|
tokens1["l"] = base_clip.tokenize(pos_l)["l"] |
|
|
|
if len(tokens1["l"]) != len(tokens1["g"]): |
|
while len(tokens1["l"]) < len(tokens1["g"]): |
|
tokens1["l"] += empty["l"] |
|
while len(tokens1["l"]) > len(tokens1["g"]): |
|
tokens1["g"] += empty["g"] |
|
|
|
cond1, pooled1 = base_clip.encode_from_tokens(tokens1, return_pooled=True) |
|
res1 = [[cond1, {"pooled_output": pooled1, "width": base_width, "height": base_height, "crop_w": crop_w, "crop_h": crop_h, "target_width": target_width, "target_height": target_height}]] |
|
|
|
|
|
tokens2 = base_clip.tokenize(neg_g) |
|
tokens2["l"] = base_clip.tokenize(neg_l)["l"] |
|
|
|
if len(tokens2["l"]) != len(tokens2["g"]): |
|
while len(tokens2["l"]) < len(tokens2["g"]): |
|
tokens2["l"] += empty["l"] |
|
while len(tokens2["l"]) > len(tokens2["g"]): |
|
tokens2["g"] += empty["g"] |
|
|
|
cond2, pooled2 = base_clip.encode_from_tokens(tokens2, return_pooled=True) |
|
res2 = [[cond2, {"pooled_output": pooled2, "width": base_width, "height": base_height, "crop_w": crop_w, "crop_h": crop_h, "target_width": target_width, "target_height": target_height}]] |
|
|
|
|
|
tokens2 = base_clip.tokenize(pos_l) |
|
tokens2["l"] = base_clip.tokenize(neg_l)["l"] |
|
|
|
if len(tokens2["l"]) != len(tokens2["g"]): |
|
while len(tokens2["l"]) < len(tokens2["g"]): |
|
tokens2["l"] += empty["l"] |
|
while len(tokens2["l"]) > len(tokens2["g"]): |
|
tokens2["g"] += empty["g"] |
|
|
|
cond2, pooled2 = base_clip.encode_from_tokens(tokens2, return_pooled=True) |
|
res3 = [[cond2, {"pooled_output": pooled2, "width": base_width, "height": base_height, "crop_w": crop_w, "crop_h": crop_h, "target_width": target_width, "target_height": target_height}]] |
|
|
|
|
|
tokens2 = base_clip.tokenize(neg_l) |
|
tokens2["l"] = base_clip.tokenize(neg_l)["l"] |
|
|
|
if len(tokens2["l"]) != len(tokens2["g"]): |
|
while len(tokens2["l"]) < len(tokens2["g"]): |
|
tokens2["l"] += empty["l"] |
|
while len(tokens2["l"]) > len(tokens2["g"]): |
|
tokens2["g"] += empty["g"] |
|
|
|
cond2, pooled2 = base_clip.encode_from_tokens(tokens2, return_pooled=True) |
|
res4 = [[cond2, {"pooled_output": pooled2, "width": base_width, "height": base_height, "crop_w": crop_w, "crop_h": crop_h, "target_width": target_width, "target_height": target_height}]] |
|
|
|
if preset == "preset A": |
|
base_positive = res1 |
|
base_negative = res2 |
|
elif preset == "preset B": |
|
base_positive = res3 |
|
base_negative = res4 |
|
elif preset == "preset C": |
|
base_positive = res1 + res3 |
|
base_negative = res2 + res4 |
|
|
|
show_help = "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki/SDXL-Nodes#cr-sdxl-base-prompt-encoder" |
|
return (base_positive, base_negative, show_help, ) |
|
|
|
|
|
|
|
|
|
|
|
''' |
|
NODE_CLASS_MAPPINGS = { |
|
"CR SDXL Style Text":CR_SDXLStyleText, |
|
"CR SDXL Base Prompt Encoder":CR_SDXLBasePromptEncoder, |
|
"CR SDXL Prompt Mix Presets":CR_PromptMixPresets, |
|
} |
|
''' |
|
|
|
|
