Implement Self-Attention Guidance (#2201)

* First SAG test

* need to put extra options on the model instead of patcher

* no errors and results seem not-broken

* Use @ashen-uncensored formula, which works better!!!

* Fix a crash when using weird resolutions. Remove an unnecessary UNet call

* Improve comments, optimize memory in blur routine

* SAG works with sampler_cfg_function
This commit is contained in:
Rafie Walker
2023-12-13 21:52:11 +01:00
committed by GitHub
parent 390078904c
commit 6761233e9d
3 changed files with 182 additions and 9 deletions

View File

@@ -1867,6 +1867,7 @@ def init_custom_nodes():
"nodes_model_downscale.py",
"nodes_images.py",
"nodes_video_model.py",
"nodes_sag.py",
]
for node_file in extras_files: