@@ -94,7 +94,7 @@ def calculate_shift(
9494 base_seq_len : int = 256 ,
9595 max_seq_len : int = 4096 ,
9696 base_shift : float = 0.5 ,
97- max_shift : float = 1.16 ,
97+ max_shift : float = 1.15 ,
9898):
9999 m = (max_shift - base_shift ) / (max_seq_len - base_seq_len )
100100 b = base_shift - m * base_seq_len
@@ -823,7 +823,7 @@ def __call__(
823823 self .scheduler .config .get ("base_image_seq_len" , 256 ),
824824 self .scheduler .config .get ("max_image_seq_len" , 4096 ),
825825 self .scheduler .config .get ("base_shift" , 0.5 ),
826- self .scheduler .config .get ("max_shift" , 1.16 ),
826+ self .scheduler .config .get ("max_shift" , 1.15 ),
827827 )
828828 timesteps , num_inference_steps = retrieve_timesteps (
829829 self .scheduler ,
@@ -993,7 +993,7 @@ def invert(
993993 self .scheduler .config .get ("base_image_seq_len" , 256 ),
994994 self .scheduler .config .get ("max_image_seq_len" , 4096 ),
995995 self .scheduler .config .get ("base_shift" , 0.5 ),
996- self .scheduler .config .get ("max_shift" , 1.16 ),
996+ self .scheduler .config .get ("max_shift" , 1.15 ),
997997 )
998998 timesteps , num_inversion_steps = retrieve_timesteps (
999999 self .scheduler ,
0 commit comments