EuuIia commited on
Commit
b641ba2
·
verified ·
1 Parent(s): 32b1b3e

Update LTX-Video/ltx_video/pipelines/pipeline_ltx_video.py

Browse files
LTX-Video/ltx_video/pipelines/pipeline_ltx_video.py CHANGED
@@ -139,8 +139,8 @@ def retrieve_timesteps(
139
  num_inference_steps: Optional[int] = None,
140
  device: Optional[Union[str, torch.device]] = None,
141
  timesteps: Optional[List[int]] = None,
142
- skip_initial_inference_steps: int = 0,
143
- skip_final_inference_steps: int = 0,
144
  **kwargs,
145
  ):
146
  """
 
139
  num_inference_steps: Optional[int] = None,
140
  device: Optional[Union[str, torch.device]] = None,
141
  timesteps: Optional[List[int]] = None,
142
+ skip_initial_inference_steps: Optional[int] = 0,
143
+ skip_final_inference_steps: Optional[int] = 0,
144
  **kwargs,
145
  ):
146
  """