Your Name commited on
Commit
e5be64b
·
1 Parent(s): 19eaaff
Files changed (2) hide show
  1. src/pipeline.py +2 -2
  2. src/scheduler_config.json +1 -1
src/pipeline.py CHANGED
@@ -10,11 +10,11 @@ from loss import SchedulerWrapper
10
  from onediffx import compile_pipe, save_pipe, load_pipe
11
 
12
  def callback_dynamic_cfg(pipe, step_index, timestep, callback_kwargs):
13
- if step_index == int(pipe.num_timesteps * 0.81):
14
  callback_kwargs['prompt_embeds'] = callback_kwargs['prompt_embeds'].chunk(2)[-1]
15
  callback_kwargs['add_text_embeds'] = callback_kwargs['add_text_embeds'].chunk(2)[-1]
16
  callback_kwargs['add_time_ids'] = callback_kwargs['add_time_ids'].chunk(2)[-1]
17
- pipe._guidance_scale = 0.10000
18
 
19
 
20
 
 
10
  from onediffx import compile_pipe, save_pipe, load_pipe
11
 
12
  def callback_dynamic_cfg(pipe, step_index, timestep, callback_kwargs):
13
+ if step_index == int(pipe.num_timesteps * 0.813):
14
  callback_kwargs['prompt_embeds'] = callback_kwargs['prompt_embeds'].chunk(2)[-1]
15
  callback_kwargs['add_text_embeds'] = callback_kwargs['add_text_embeds'].chunk(2)[-1]
16
  callback_kwargs['add_time_ids'] = callback_kwargs['add_time_ids'].chunk(2)[-1]
17
+ pipe._guidance_scale = 0.10001
18
 
19
 
20
 
src/scheduler_config.json CHANGED
@@ -1,3 +1,3 @@
1
  {
2
- "_by": "ColdAsIce"
3
  }
 
1
  {
2
+ "_by": "Givemeaname"
3
  }