mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-13 23:53:02 +00:00
Kasisnu
48 lines
941 B
Python
48 lines
941 B
Python
import sys
|
|
import os
|
|
import torch
|
|
from PIL import Image, ImageDraw, ImageOps
|
|
import math
|
|
import numpy as np
|
|
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "comfy"))
|
|
|
|
|
|
class YaeCombineConditionings:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {"required": {
|
|
"conditioning1": ("CONDITIONING", ),
|
|
"conditioning2": ("CONDITIONING", ),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("CONDITIONING", )
|
|
FUNCTION = "doit"
|
|
|
|
CATEGORY = "ImpactPack/Util"
|
|
|
|
def doit(self, **kwargs):
|
|
res = []
|
|
for k, v in kwargs.items():
|
|
if v != 0:
|
|
res += v
|
|
|
|
return (res, )
|
|
|
|
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"YaeCombineConditionings": YaeCombineConditionings
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"YaeCombineConditionings": "Yae's Combine Conditionings"
|
|
}
|
|
|
|
|
|
|
|
|