Update __init__.py

2024-05-15 06:08:14 +02:00
parent a776316fda
commit c3376eed03
1 changed files with 1 additions and 0 deletions
@@ -9,6 +9,7 @@ NODE_CLASS_MAPPINGS = {
    "Automatic CFG - Excellent attention": simpleDynamicCFGExcellentattentionPatch,
    "Automatic CFG - Advanced": advancedDynamicCFG,
    "Automatic CFG - Post rescale only": postCFGrescaleOnly,
+    "Automatic CFG - Custom attentions": simpleDynamicCFGCustomAttentionPatch,
    "Automatic CFG - Attention modifiers": attentionModifierParametersNode,
    "Automatic CFG - Attention modifiers tester": attentionModifierBruteforceParametersNode,
    "Automatic CFG - Unpatch function": simpleDynamicCFGunpatch,