34 Commits
Author SHA1 Message Date
blepping 8de02bc769 Workaround for recent ComfyUI frontend breakage 2025-05-06 04:33:59 -06:00
blepping 54d60e0d18 Different approach to integrating external modules (#28)
* Different approach to integrating external modules
* Other internal cleanups
2024-12-24 21:47:01 -07:00
blepping 30ffe2d939 Make block in MSW-MSA attention warnings more user friendly 2024-10-12 08:21:56 -06:00
blepping debeccf722 Fix compatibility with pre-Python 3.11.
Use enums a bit more responsibly.

Make the verbose config field an integer verbosity level.
2024-10-12 05:50:25 -06:00
blepping e62bd8ea41 Allow applying MSW-MSA attention to attn2 (not a good idea though).
Move MSW-MSA attention scaling modes into advanced YAML options.
2024-10-10 17:10:44 -06:00
blepping 88dd9340aa Add advanced YAML parameters input to normal nodes.
Allow fading out CA downscale.

Use a pixel increment when scaling to ensure more compatible sizes.

Allow multiplying tensors on input/out for both RAUNet and MSW-MSA attention.

Add an auto model type for simple nodes and try to guess the model.

Allow CA scaling to use different factors for width and height.

Default CA downscaling method is now adaptive_avg_pool2d (and also add that scaling type).

CA input block scaling can now apply after the skip connection like Kohya Deep Shrink.

Various refactoring and cleanups.
2024-10-10 09:05:45 -06:00
Pam eca9689c03 Add upscale mode selector for MSW MSA 2024-09-30 18:03:53 +05:00
Pam c79cc33ff9 Rewrite downsample_ratio calculation 2024-09-29 11:08:17 +05:00
Pam f17b2efe23 update msw msa
change SDXL default values
2024-09-27 22:19:13 +05:00
blepping 2b3fadfaf2 Try to fix issue with RAUNet model patching changing seeds
Better tooltips
2024-08-27 22:07:02 -06:00
blepping 2b9c5b1c2e ComfyUI-GGUF compatibility for RAUNet 2024-08-25 16:46:34 -06:00
blepping d4d3fd0ff3 Fix issue with controlnet workaround 2024-08-17 01:34:02 -06:00
blepping 4f48873f98 Make up/downsample block targeting more resiliant in RAUNet 2024-08-15 19:06:56 -06:00
blepping 4e8ef65a7e Slight consistency tweak for tooltip text. 2024-08-15 02:39:49 -06:00
blepping c1f7e80f78 Fix output blocks tooltip in ApplyRAUNet node 2024-08-14 10:56:44 -06:00
blepping 00e41dc6b1 Add tooltips metadata to nodes 2024-08-14 10:54:24 -06:00
blepping f0f38f4288 Fix apply_control patch getting applied multiple times 2024-08-13 03:54:35 -06:00
blepping b9cbf88475 Fix simple nodes
Update docs and examples

Add changelog
2024-08-13 03:48:29 -06:00
blepping c95d815c7f Attempt to fix RAUNet duplicate object patch issue 2024-08-10 11:16:04 -06:00
blepping 8d329808a7 Be more consistent in MSW-MSA window code 2024-08-10 06:52:40 -06:00
blepping aad095ab14 Refactor to avoid having to monkeypatch forward_timestep_embed
Other cleanups

Allow setting a downscale factor and mode for CA downsampling

Add more type annotations

Make it so MSW-MSA attention failing is a warning rather than hard error

Adjust a few Ruff lint settings
2024-08-10 06:27:59 -06:00
Pam 8de0f98c3d Move two_stage_upscale toggle into two_stage_upscale_mode 2024-08-05 17:54:47 +05:00
Pam 5636c8c514 Refactor RAUNet code 2024-08-04 23:57:59 +05:00
Pam 0d762c1e16 Revert two_stage_upscale rename 2024-08-02 14:55:39 +05:00
Pam 19baa1626c Fix two_stage_upscale assignment 2024-08-02 08:11:27 +05:00
blepping 6111b81df3 Add an experimental ControlNet workaround 2024-05-21 09:58:43 -06:00
blepping 02666ceff4 More reliable ComfyUI-bleh integration (hopefully) 2024-05-21 09:58:15 -06:00
blepping e54ff4d58d Remove match/case statements for Python 3.9 compatibility 2024-05-09 22:18:45 -06:00
blepping 0eba522c7d Add simple node versions
Update documentation

Make resolution mismatch error in MSW-MSA attention more friendly
2024-05-07 03:48:55 -06:00
blepping 16abd6868d Fix incorrect default setting for ApplyMSWMSAAttention node (block 0 isn't attention)
Some refactoring/cleanups

Update documentation with use cases/compatibility notes
2024-05-07 00:49:08 -06:00
blepping fc14c0fafb Make timestep forward more robust (should fix #3) 2024-04-29 08:20:44 -06:00
blepping 2cc4529472 Allow use of bleh upscale methods when possible 2024-04-27 18:13:07 -06:00
blepping e693373a12 Detach and move sigmas to CPU before comparing
Clean up/refactor activation checking code
2024-04-27 14:03:29 -06:00
blepping f7fea53036 Initial 2024-04-27 08:04:24 -06:00