|
24 | 24 | LoRAExtractFixed, LoRAExtractRatio, LoRAExtractQuantile, |
25 | 25 | LoRAExtractKnee, LoRAExtractFrobenius |
26 | 26 | ) |
| 27 | +from .nodes.dora_extract_wd import ( |
| 28 | + DoRAExtractFixed, DoRAExtractRatio, DoRAExtractQuantile, |
| 29 | + DoRAExtractKnee, DoRAExtractFrobenius |
| 30 | +) |
27 | 31 | from .nodes.lora_resize import ( |
28 | 32 | LoRAResizeFixed, LoRAResizeRatio, |
29 | 33 | LoRAResizeFrobenius, LoRAResizeCumulative, |
@@ -61,6 +65,9 @@ async def get_node_list(self) -> list[type[io.ComfyNode]]: |
61 | 65 | # LoRA Extraction |
62 | 66 | LoRAExtractFixed, LoRAExtractRatio, LoRAExtractQuantile, |
63 | 67 | LoRAExtractKnee, LoRAExtractFrobenius, |
| 68 | + # DoRA Extraction |
| 69 | + DoRAExtractFixed, DoRAExtractRatio, DoRAExtractQuantile, |
| 70 | + DoRAExtractKnee, DoRAExtractFrobenius, |
64 | 71 | # LoRA Resize |
65 | 72 | LoRAResizeFixed, LoRAResizeRatio, |
66 | 73 | LoRAResizeFrobenius, LoRAResizeCumulative, |
|
0 commit comments