-
Notifications
You must be signed in to change notification settings - Fork 4
/
__init__.py
40 lines (33 loc) · 1.12 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
from .py import settings
BLEH_VERSION = 1
settings.load_settings()
if settings.SETTINGS.btp_enabled:
from .py import betterTaesdPreview # noqa: F401
from .py.nodes import (
deepShrink,
hyperTile,
misc,
modelPatchConditional,
ops,
refinerAfter,
samplers,
)
NODE_CLASS_MAPPINGS = {
"BlehBlockOps": ops.BlehBlockOps,
"BlehDeepShrink": deepShrink.DeepShrinkBleh,
"BlehDiscardPenultimateSigma": misc.DiscardPenultimateSigma,
"BlehDisableNoise": misc.BlehDisableNoise,
"BlehPlug": misc.BlehPlug,
"BlehForceSeedSampler": samplers.BlehForceSeedSampler,
"BlehHyperTile": hyperTile.HyperTileBleh,
"BlehInsaneChainSampler": samplers.BlehInsaneChainSampler,
"BlehLatentScaleBy": ops.BlehLatentScaleBy,
"BlehLatentOps": ops.BlehLatentOps,
"BlehModelPatchConditional": modelPatchConditional.ModelPatchConditionalNode,
"BlehRefinerAfter": refinerAfter.BlehRefinerAfter,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"BlehHyperTile": "HyperTile (bleh)",
"BlehDeepShrink": "Kohya Deep Shrink (bleh)",
}
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "BLEH_VERSION"]