40 lines
998 B
Python
40 lines
998 B
Python
from nodes import KSampler
|
|
from .fabric.fabric import fabric_sample
|
|
|
|
|
|
class KSamplerFABRIC:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
inputs = KSampler.INPUT_TYPES()
|
|
added_inputs = {
|
|
"required": {
|
|
"null_pos": ("CONDITIONING",),
|
|
"null_neg": ("CONDITIONING",),
|
|
},
|
|
"optional": {
|
|
"pos_latents": ("LATENT",),
|
|
"neg_latents": ("LATENT",),
|
|
}
|
|
}
|
|
inputs["required"].update(added_inputs["required"])
|
|
if "optional" not in inputs:
|
|
inputs["optional"] = {}
|
|
inputs["optional"].update(added_inputs["optional"])
|
|
return inputs
|
|
|
|
RETURN_TYPES = ("LATENT",)
|
|
FUNCTION = "sample"
|
|
CATEGORY = "FABRIC"
|
|
|
|
def sample(self, *args, **kwargs):
|
|
return fabric_sample(*args, **kwargs)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"KSamplerFABRIC": KSamplerFABRIC,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"KSamplerFABRIC": "KSampler With FABRIC",
|
|
}
|