ab

src/pipeline.py

@@ -10,7 +10,7 @@ from loss import SchedulerWrapper
 from onediffx import compile_pipe, save_pipe, load_pipe
 
 def callback_dynamic_cfg(pipe, step_index, timestep, callback_kwargs):
-  if step_index == int(pipe.num_timesteps * 0.75):
+  if step_index == int(pipe.num_timesteps * 0.81):
     callback_kwargs['prompt_embeds'] = callback_kwargs['prompt_embeds'].chunk(2)[-1]
     callback_kwargs['add_text_embeds'] = callback_kwargs['add_text_embeds'].chunk(2)[-1]
     callback_kwargs['add_time_ids'] = callback_kwargs['add_time_ids'].chunk(2)[-1]