mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-02-28 22:51:45 +00:00
6c9bd11fa3
* Add 'sigmas' to transformer_options so that downstream code can know about the full scope of current sampling run, fix Hook Keyframes' guarantee_steps=1 inconsistent behavior with sampling split across different Sampling nodes/sampling runs by referencing 'sigmas' * Cleaned up hooks.py, refactored Hook.should_register and add_hook_patches to use target_dict instead of target so that more information can be provided about the current execution environment if needed * Refactor WrapperHook into TransformerOptionsHook, as there is no need to separate out Wrappers/Callbacks/Patches into different hook types (all affect transformer_options) * Refactored HookGroup to also store a dictionary of hooks separated by hook_type, modified necessary code to no longer need to manually separate out hooks by hook_type * In inner_sample, change "sigmas" to "sampler_sigmas" in transformer_options to not conflict with the "sigmas" that will overwrite "sigmas" in _calc_cond_batch * Refactored 'registered' to be HookGroup instead of a list of Hooks, made AddModelsHook operational and compliant with should_register result, moved TransformerOptionsHook handling out of ModelPatcher.register_all_hook_patches, support patches in TransformerOptionsHook properly by casting any patches/wrappers/hooks to proper device at sample time * Made hook clone code sane, made clear ObjectPatchHook and SetInjectionsHook are not yet operational * Fix performance of hooks when hooks are appended via Cond Pair Set Props nodes by properly caching between positive and negative conds, make hook_patches_backup behave as intended (in the case that something pre-registers WeightHooks on the ModelPatcher instead of registering it at sample time) * Filter only registered hooks on self.conds in CFGGuider.sample * Make hook_scope functional for TransformerOptionsHook * removed 4 whitespace lines to satisfy Ruff, * Add a get_injections function to ModelPatcher * Made TransformerOptionsHook contribute to registered hooks properly, added some doc strings and removed a so-far unused variable * Rename AddModelsHooks to AdditionalModelsHook, rename SetInjectionsHook to InjectionsHook (not yet implemented, but at least getting the naming figured out) * Clean up a typehint |
||
|---|---|---|
| .. | ||
| chainner_models | ||
| nodes_advanced_samplers.py | ||
| nodes_align_your_steps.py | ||
| nodes_attention_multiply.py | ||
| nodes_audio.py | ||
| nodes_canny.py | ||
| nodes_clip_sdxl.py | ||
| nodes_compositing.py | ||
| nodes_cond.py | ||
| nodes_controlnet.py | ||
| nodes_cosmos.py | ||
| nodes_custom_sampler.py | ||
| nodes_differential_diffusion.py | ||
| nodes_flux.py | ||
| nodes_freelunch.py | ||
| nodes_gits.py | ||
| nodes_hooks.py | ||
| nodes_hunyuan.py | ||
| nodes_hypernetwork.py | ||
| nodes_hypertile.py | ||
| nodes_images.py | ||
| nodes_ip2p.py | ||
| nodes_latent.py | ||
| nodes_load_3d.py | ||
| nodes_lora_extract.py | ||
| nodes_lt.py | ||
| nodes_mahiro.py | ||
| nodes_mask.py | ||
| nodes_mochi.py | ||
| nodes_model_advanced.py | ||
| nodes_model_downscale.py | ||
| nodes_model_merging_model_specific.py | ||
| nodes_model_merging.py | ||
| nodes_morphology.py | ||
| nodes_pag.py | ||
| nodes_perpneg.py | ||
| nodes_photomaker.py | ||
| nodes_pixart.py | ||
| nodes_post_processing.py | ||
| nodes_rebatch.py | ||
| nodes_sag.py | ||
| nodes_sd3.py | ||
| nodes_sdupscale.py | ||
| nodes_slg.py | ||
| nodes_stable3d.py | ||
| nodes_stable_cascade.py | ||
| nodes_tomesd.py | ||
| nodes_torch_compile.py | ||
| nodes_upscale_model.py | ||
| nodes_video_model.py | ||
| nodes_webcam.py | ||