mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-25 08:40:31 +00:00
prep for new workflows by yae
This commit is contained in:
@@ -0,0 +1,47 @@
|
||||
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"
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user