Skip to content

Commit 1a41958

Browse files
authored
feat(diffusers/pipelines): add LTXConditionPipeline (#997)
1 parent 9d641f5 commit 1a41958

File tree

11 files changed

+1825
-80
lines changed

11 files changed

+1825
-80
lines changed

docs/diffusers/api/pipelines/ltx_video.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ Available models:
2626
|:----------------------------------------------------------------------------------------------------------:|:-----------------:|
2727
| [`LTX Video 0.9.0`](https://huggingface.co/Lightricks/LTX-Video/blob/main/ltx-video-2b-v0.9.safetensors) | `ms.bfloat16` |
2828
| [`LTX Video 0.9.1`](https://huggingface.co/Lightricks/LTX-Video/blob/main/ltx-video-2b-v0.9.1.safetensors) | `ms.bfloat16` |
29+
| [`LTX Video 0.9.5`](https://huggingface.co/Lightricks/LTX-Video/blob/main/ltx-video-2b-v0.9.5.safetensors) | `ms.bfloat16` |
2930

3031
Note: The recommended dtype is for the transformer component. The VAE and text encoders can be either `ms.float32`, `ms.bfloat16` or `ms.float16` but the recommended dtype is `ms.bfloat16` as used in the original repository.
3132

@@ -101,4 +102,6 @@ Refer to [this section](https://mindspore-lab.github.io/mindone/latest/diffusers
101102

102103
::: mindone.diffusers.LTXImageToVideoPipeline
103104

105+
::: mindone.diffusers.LTXConditionPipeline
106+
104107
::: mindone.diffusers.pipelines.ltx.pipeline_output.LTXPipelineOutput

mindone/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,7 @@
178178
"LDMTextToImagePipeline",
179179
"LEditsPPPipelineStableDiffusion",
180180
"LEditsPPPipelineStableDiffusionXL",
181+
"LTXConditionPipeline",
181182
"LTXImageToVideoPipeline",
182183
"LTXPipeline",
183184
"Lumina2Pipeline",
@@ -484,6 +485,7 @@
484485
LDMTextToImagePipeline,
485486
LEditsPPPipelineStableDiffusion,
486487
LEditsPPPipelineStableDiffusionXL,
488+
LTXConditionPipeline,
487489
LTXImageToVideoPipeline,
488490
LTXPipeline,
489491
Lumina2Pipeline,

0 commit comments

Comments
 (0)