Skip to content

Commit 3a505e7

Browse files
committed
Add work in progress experimental learned extraction nodes
1 parent 1512fa7 commit 3a505e7

2 files changed

Lines changed: 797 additions & 0 deletions

File tree

__init__.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,10 @@
2828
DoRAExtractFixed, DoRAExtractRatio, DoRAExtractQuantile,
2929
DoRAExtractKnee, DoRAExtractFrobenius
3030
)
31+
from .nodes.dora_learned_wd import (
32+
DoRALearnedExtractFixed, DoRALearnedExtractRatio, DoRALearnedExtractQuantile,
33+
DoRALearnedExtractKnee, DoRALearnedExtractFrobenius
34+
)
3135
from .nodes.lora_resize import (
3236
LoRAResizeFixed, LoRAResizeRatio,
3337
LoRAResizeFrobenius, LoRAResizeCumulative,
@@ -68,6 +72,9 @@ async def get_node_list(self) -> list[type[io.ComfyNode]]:
6872
# DoRA Extraction
6973
DoRAExtractFixed, DoRAExtractRatio, DoRAExtractQuantile,
7074
DoRAExtractKnee, DoRAExtractFrobenius,
75+
# Learned DoRA Extraction
76+
DoRALearnedExtractFixed, DoRALearnedExtractRatio, DoRALearnedExtractQuantile,
77+
DoRALearnedExtractKnee, DoRALearnedExtractFrobenius,
7178
# LoRA Resize
7279
LoRAResizeFixed, LoRAResizeRatio,
7380
LoRAResizeFrobenius, LoRAResizeCumulative,

0 commit comments

Comments
 (0)