Fix for TypeError: teacache_flux_forward() got an unexpected keyword argument 'timestep_zero_index'

[nickbrown1968]

Commit 44f1246c899ed188759f799dbd00c31def289114 of Comfyui appeared to add Flux2 support. In doing so a timestep_zero_index parameter was added to forward_orig() in the flux model (in comfy/ldm/flux/model.py). This broke teacache. I resolved this by adding this parameter to teacache_flux_forward() in ComfyUI\custom_nodes\teacache\nodes.py:

def teacache_flux_forward(
        self,
        img: Tensor,
        img_ids: Tensor,
        txt: Tensor,
        txt_ids: Tensor,
        timesteps: Tensor,
        y: Tensor,
        guidance: Tensor = None,
        control = None,
        timestep_zero_index=None,
        transformer_options={},
        attn_mask: Tensor = None,
    )

Hope this helps someone.