【ComfyUI】ComfyUI-AnimateDiff-Evolved
ComfyUI-AnimateDiff-Evolved
挖到comfyui的animateDiff实现底层去了,试图找到animate diff如何用for循环,每次只处理连续的16个frames而不是所有
已经找到了ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/animatediff/nodes_gen2.py
本以为应该是做了个model wrapper,调用wrapper的forward或者__call__的时候进行for循环,结果发现根本不是
这个文件ComfyUI/custom_nodes/ComfyUI-AnimateDiff-Evolved/animatediff/model_injection.py
几乎只是提供了一个模型包装,没有任何的forward或者__call__实现,而真正的循环处理在ComfyUI/comfy/samplers.py
,找到里面的calc_cond_batch
函数,可以发现这个for循环的实现方式是通过这个函数做的