Add prepare_sampling wrapper allowing custom nodes to more accurately report noise_shape (#7500)

2025-04-09 21:43:35 +08:00
parent ab31b64412
commit e346d8584e
2 changed files with 8 additions and 0 deletions
--- a/comfy/patcher_extension.py
+++ b/comfy/patcher_extension.py
@@ -48,6 +48,7 @@ def get_all_callbacks(call_type: str, transformer_options: dict, is_model_option

 class WrappersMP:
    OUTER_SAMPLE = "outer_sample"
+    PREPARE_SAMPLING = "prepare_sampling"
    SAMPLER_SAMPLE = "sampler_sample"
    CALC_COND_BATCH = "calc_cond_batch"
    APPLY_MODEL = "apply_model"