Added initial support for finetuning wan i2v WIP

2026-04-28 10:11:14 +00:00 · 2025-04-07 20:34:38 -06:00
parent 38ad5a4644
commit a8680c75eb
10 changed files with 575 additions and 286 deletions
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -1831,6 +1831,7 @@ class StableDiffusion:
            return_conditional_pred=False,
            guidance_embedding_scale=1.0,
            bypass_guidance_embedding=False,
+            batch: Union[None, 'DataLoaderBatchDTO'] = None,
            **kwargs,
    ):
        conditional_pred = None