Add missing model_options param

This commit is contained in:
Jedrzej Kosinski 2024-12-31 11:37:22 -06:00
parent 67758f50f3
commit 0af314cb86

View File

@ -224,7 +224,7 @@ def _calc_cond_batch(model: 'BaseModel', conds: list[list[dict]], x_in: torch.Te
default_conds.append(default_c)
if has_default_conds:
finalize_default_conds(model, hooked_to_run, default_conds, x_in, timestep)
finalize_default_conds(model, hooked_to_run, default_conds, x_in, timestep, model_options)
model.current_patcher.prepare_state(timestep)