Spaces:
Paused
Paused
Update api/ltx/vae_aduc_pipeline.py
Browse files
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
|
| 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[
|
| 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,
|