delete useless nodes and change to use primitive node

This commit is contained in:
hnmr293
2023-03-30 22:07:06 +09:00
parent 0b4b821acb
commit 4210536ffd
8 changed files with 6 additions and 161 deletions
-16
View File
@@ -2,22 +2,6 @@
![](./nodes.png)
## Basic nodes
|category|node name|input type|output type|desc.|
| --- | --- | --- | --- | --- |
|value|Text|`STRING`|`TEXT`| - |
|value|Integer|`INT`|`Integer`| - |
|value|Float|`FLOAT`|`Float`| - |
|value|SamplerName|(dropdown)|`SamplerName`| - |
|value|SchedulerName|(dropdown)|`SchedulerName`| - |
## CLIP nodes
|category|node name|input type|output type|desc.|
| --- | --- | --- | --- | --- |
|conditioning|CLIPTextEncode2|`CLIP`, `TEXT`|`CONDITIONING`|external prompt input|
## Latent nodes
|category|node name|input type|output type|desc.|
-26
View File
@@ -1,7 +1,3 @@
from .text import Text
from .number import Integer, Float
from .sampler import SamplerName, SchedulerName
from .cliptextencode2 import CLIPTextEncode2
from .randomlatent import RandomLatentImage
from .vae import VAEDecodeBatched, VAEEncodeBatched
from .sample import KSamplerSetting, KSamplerOverrided, KSamplerXYZ
@@ -9,28 +5,6 @@ from .model import StateDictLoader, Dict2Model, StateDictMerger, StateDictMerger
from .image import GridImage
NODE_CLASS_MAPPINGS = {
# basic nodes
## text output
'Text': Text,
## integer output
'Integer': Integer,
## float output
'Float': Float,
## sampler selection
'SamplerName': SamplerName,
## scheduler selection
'SchedulerName': SchedulerName,
# conditioning
## same as CLIPTextEncode, but the prompt and CLIP are external inputs
'CLIPTextEncode2': CLIPTextEncode2,
# latent
'RandomLatentImage': RandomLatentImage,
-11
View File
@@ -1,11 +0,0 @@
class CLIPTextEncode2:
@classmethod
def INPUT_TYPES(cls):
return {"required": {"clip": ("CLIP", ), "text": ("TEXT",)}}
RETURN_TYPES = ("CONDITIONING",)
FUNCTION = "encode"
CATEGORY = "conditioning"
def encode(self, clip, text):
return ([[clip.encode(text), {}]], )
+1 -1
View File
@@ -279,7 +279,7 @@ class StateDictMergerBlockWeighted(StateDictMerger):
d['required']['base_alpha'] = d['required']['alpha']
del d['required']['alpha']
del d['optional']['model_C']
d['required']['alphas'] = ('TEXT',)
d['required']['alphas'] = ('STRING', { 'multiline': True, 'default': '' })
return d
def execute(
-43
View File
@@ -1,43 +0,0 @@
import re
re_range = re.compile(r"\s*([+-]?\s*\d+)\s*-\s*([+-]?\s*\d+)(?:\s*\(([+-]\d+)\s*\))?\s*")
re_range_float = re.compile(r"\s*([+-]?\s*\d+(?:.\d*)?)\s*-\s*([+-]?\s*\d+(?:.\d*)?)(?:\s*\(([+-]\d+(?:.\d*)?)\s*\))?\s*")
class Integer:
@classmethod
def INPUT_TYPES(cls):
return {
'required': {
'value': ('INT', {})
}
}
RETURN_TYPES = ('Integer',)
FUNCTION = 'execute'
CATEGORY = 'value'
def execute(self, value: int):
return (value,)
class Float:
@classmethod
def INPUT_TYPES(cls):
return {
'required': {
'value': ('FLOAT', {})
}
}
RETURN_TYPES = ('Float',)
FUNCTION = 'execute'
CATEGORY = 'value'
def execute(self, value: float):
return (value,)
+5 -5
View File
@@ -234,11 +234,11 @@ class KSamplerXYZ:
},
'optional': {
'model': ('MODEL',),
'seed': ('TEXT',),
'steps': ('TEXT',),
'cfg': ('TEXT',),
'sampler_name': ('TEXT',),
'scheduler': ('TEXT', ),
'seed': ('STRING', { 'multiline': True, 'default': '' }),
'steps': ('STRING', { 'multiline': True, 'default': '' }),
'cfg': ('STRING', { 'multiline': True, 'default': '' }),
'sampler_name': ('STRING', { 'multiline': True, 'default': '' }),
'scheduler': ('STRING', { 'multiline': True, 'default': '' }),
}
}
-39
View File
@@ -1,39 +0,0 @@
import comfy.samplers
class SamplerName:
@classmethod
def INPUT_TYPES(cls):
return {
'required': {
'value': (comfy.samplers.KSampler.SAMPLERS,)
}
}
RETURN_TYPES = ('SamplerName',)
FUNCTION = 'execute'
CATEGORY = 'value'
def execute(self, value: str):
return (value,)
class SchedulerName:
@classmethod
def INPUT_TYPES(cls):
return {
'required': {
'value': (comfy.samplers.KSampler.SCHEDULERS,)
}
}
RETURN_TYPES = ('SchedulerName',)
FUNCTION = 'execute'
CATEGORY = 'value'
def execute(self, value: str):
return (value,)
-20
View File
@@ -1,20 +0,0 @@
class Text:
@classmethod
def INPUT_TYPES(cls):
return {
'required': {
'text': ('STRING', {
'multiline': True,
})
}
}
RETURN_TYPES = ('TEXT',) # currently no widgets can receive 'STRING' inputs...
FUNCTION = 'execute'
CATEGORY = 'value'
def execute(self, text: str):
return (text,)