Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
338 changes: 337 additions & 1 deletion src/diffusers/pipelines/ltx2/export_utils.py

Large diffs are not rendered by default.

315 changes: 301 additions & 14 deletions src/diffusers/pipelines/ltx2/image_processor.py

Large diffs are not rendered by default.

445 changes: 383 additions & 62 deletions src/diffusers/pipelines/ltx2/pipeline_ltx2_hdr_lora.py

Large diffs are not rendered by default.

1 change: 1 addition & 0 deletions src/diffusers/utils/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,6 +97,7 @@
is_nvidia_modelopt_version,
is_onnx_available,
is_opencv_available,
is_openexr_available,
is_optimum_quanto_available,
is_optimum_quanto_version,
is_outlines_available,
Expand Down
5 changes: 5 additions & 0 deletions src/diffusers/utils/import_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -220,6 +220,7 @@ def _is_package_available(pkg_name: str, get_dist_name: bool = False) -> tuple[b
_sdnq_available, _sdnq_version = _is_package_available("sdnq")
_flashpack_available, _flashpack_version = _is_package_available("flashpack")
_av_available, _av_version = _is_package_available("av")
_openexr_available, _openexr_version = _is_package_available("OpenEXR")


def is_torch_available():
Expand Down Expand Up @@ -422,6 +423,10 @@ def is_av_available():
return _av_available


def is_openexr_available():
return _openexr_available


# docstyle-ignore
INFLECT_IMPORT_ERROR = """
{0} requires the inflect library but it was not found in your environment. You can install it with pip: `pip install
Expand Down
2 changes: 1 addition & 1 deletion tests/pipelines/ltx2/test_ltx2_hdr.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ class LTX2HDRPipelineTesterConfig(LTX2BaseTesterConfig):
optional_input_params = frozenset(
["num_inference_steps", "num_videos_per_prompt", "generator", "latents", "output_type", "return_dict"]
)
unset_components = ("audio_scheduler",)
unset_components = ("audio_scheduler", "diffusion_decoder")

def get_dummy_inputs(self):
generator = self.get_generator(0)
Expand Down
470 changes: 470 additions & 0 deletions tests/pipelines/ltx2/test_ltx2_hdr_color.py

Large diffs are not rendered by default.

387 changes: 387 additions & 0 deletions tests/pipelines/ltx2/test_ltx2_hdr_sdr_to_hdr.py

Large diffs are not rendered by default.

Loading