49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
from .randomlatent import RandomLatentImage
|
|
from .vae import VAEDecodeBatched, VAEEncodeBatched
|
|
from .sample import KSamplerSetting, KSamplerOverrided, KSamplerXYZ
|
|
from .model import StateDictLoader, Dict2Model, StateDictMerger, StateDictMergerBlockWeighted
|
|
from .image import GridImage
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
# latent
|
|
|
|
'RandomLatentImage': RandomLatentImage,
|
|
|
|
## pass latents to VAE separately
|
|
'VAEDecodeBatched': VAEDecodeBatched,
|
|
'VAEEncodeBatched': VAEEncodeBatched,
|
|
|
|
# sampling
|
|
|
|
## put parameters for sampler into a dict
|
|
'KSamplerSetting': KSamplerSetting,
|
|
|
|
## KSampler with a dict as default setting
|
|
'KSamplerOverrided': KSamplerOverrided,
|
|
|
|
## XYZ plotting
|
|
'KSamplerXYZ': KSamplerXYZ,
|
|
|
|
# loader
|
|
|
|
## loads state_dict of the specified checkpoint and returns it
|
|
'StateDictLoader': StateDictLoader,
|
|
|
|
# model
|
|
|
|
## creates model from state_dict loaded by `StateDictLoader`
|
|
'Dict2Model': Dict2Model,
|
|
|
|
## merge two (weighted sum) or three (add difference) state_dict
|
|
'StateDictMerger': StateDictMerger,
|
|
|
|
## merge block weighted
|
|
## weights should be specified by Text
|
|
'StateDictMergerBlockWeighted': StateDictMergerBlockWeighted,
|
|
|
|
# image
|
|
|
|
## rearrange images to single image with specified columns and gap
|
|
'GridImage': GridImage,
|
|
}
|