wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-16 08:22:36 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

Nodes can now patch the unet function.

Browse Source
This commit is contained in:
comfyanonymous 2023-07-22 17:01:12 -04:00
parent 676b635914
commit 4d83ca88a5
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/samplers.py
View File

@ -248,6 +248,9 @@ def sampling_function(model_function, x, timestep, uncond, cond, cond_scale, con
c['transformer_options'] = transformer_options c['transformer_options'] = transformer_options
if 'model_function_wrapper' in model_options:
output = model_options['model_function_wrapper'](model_function, {"input": input_x, "timestep": timestep_, "c": c, "cond_or_uncond": cond_or_uncond}).chunk(batch_chunks)
else:
output = model_function(input_x, timestep_, **c).chunk(batch_chunks) output = model_function(input_x, timestep_, **c).chunk(batch_chunks)
del input_x del input_x