eeuuia commited on
Commit
1fedca7
·
verified ·
1 Parent(s): 64591a0

Update api/ltx/vae_aduc_pipeline.py

Browse files
Files changed (1) hide show
  1. api/ltx/vae_aduc_pipeline.py +2 -2
api/ltx/vae_aduc_pipeline.py CHANGED
@@ -46,7 +46,7 @@ if str(LTX_VIDEO_REPO_DIR.resolve()) not in sys.path:
46
 
47
  from ltx_video.models.autoencoders.vae_encode import vae_encode, vae_decode, latent_to_pixel_coords
48
  from ltx_video.models.autoencoders.causal_video_autoencoder import CausalVideoAutoencoder
49
- from ltx_video.pipelines.pipeline_ltx_video import PipelineConditioningItem
50
 
51
 
52
  @dataclass
@@ -205,7 +205,7 @@ class VaeAducPipeline:
205
  @log_function_io
206
  def prepare_conditioning(
207
  self,
208
- conditioning_items: Optional[List[Union[PipelineConditioningItem, LatentConditioningItem]]],
209
  init_latents: torch.Tensor,
210
  num_frames: int,
211
  height: int,
 
46
 
47
  from ltx_video.models.autoencoders.vae_encode import vae_encode, vae_decode, latent_to_pixel_coords
48
  from ltx_video.models.autoencoders.causal_video_autoencoder import CausalVideoAutoencoder
49
+ from ltx_video.pipelines.pipeline_ltx_video import ConditioningItem
50
 
51
 
52
  @dataclass
 
205
  @log_function_io
206
  def prepare_conditioning(
207
  self,
208
+ conditioning_items: Optional[List[Union[ConditioningItem, LatentConditioningItem]]],
209
  init_latents: torch.Tensor,
210
  num_frames: int,
211
  height: int,