Create a new SDOps instance with the specified prefix, suffix and contains matching added to the mapping.

Create a new SDOps instance with the specified replacement added to the mapping.

Load state dict from path or paths (for sharded model storage) and apply sd_ops

Run a single fused Gemma forward over a batch of prompts. Calls the inner model (self.model.model) to skip lm_head logits computation

Blend denoised output with clean state based on mask.

Record per-element losses into their sigma buckets. Args: sigmas: Per-element sigma values, one per batch element. los

Create a list of conditionings by replacing the latent at the first frame with the encoded image if present and using other encoded images as the

Decode all latent files in the directory recursively. Args: latents_dir: Directory containing latent files (.pt) outpu

Split along the batch dimension into chunks of the given sizes.

Format any value for display.

Patchify the latent state.

Context manager that yields a model and releases its memory on exit. Moves all parameters and buffers to ``meta`` device on exit, which immedi

Convert device specification to torch.device.

Create a 2D convolution layer that can be either causal or non-causal. Args: in_channels: Number of input channels out_channe

Assert that the resolution is divisible by the required divisor. For two-stage pipelines, the resolution must be divisible by 64. For one-stag

Detect pipeline params from checkpoint metadata. Reads the ``model_version`` field from the safetensors config metadata. Returns ``LTX_2_3_PAR

Get the number of video chunks for a given number of frames and tiling configuration. Args: num_frames: Number of frames in the video

Create a new SDOps instance with the specified value operation added to the mapping.

Load file paths from a dataset column, resolving relative to the dataset file's directory.

Create or return a MultiModalGuiderFactory. Pass constant params for a single-params factory (uses MultiModalGuiderFactory.constant), or an e

Pre-parse argv to extract the checkpoint path before building the full parser.

Load frames from a video file using PyAV. Args: video_path: Path to the video file max_frames: Maximum number of frames to read. I

Relative path used to name a sample's cached output, mirroring the input layout. Normally media lives under the dataset directory and this is just

Apply the conditioning to the latent state. Args: latent_state: The latent state to apply the conditioning to. This is st

Byte size of one contiguous, 16-byte-aligned buffer holding *layout* (>= 1).

Load a state dict from disk, using registry caching.

Resolve to a concrete callable. Use this at module init time so that torch.compile can trace through the attention call without graph breaks.

Save to ``out`` atomically via per-PID temp file + replace. Crash mid-write leaves an orphan ``.tmp.<pid>`` file that the skip logic ignores.

Generate video position embeddings using ltx_core's native implementation. Args: num_frames: Number of latent frames h

Pick the memory format for a workspace allocation. When ``prefer_channels_last_3d`` is True and ``t`` is 5D, return ``channels_last_3d`` regar

Build a single-backend ``PytorchAttention`` pin, raising if the backend can't actually serve the call on this machine. Used by both :meth:`Att

Create a normalization layer based on the normalization type. Args: in_channels: Number of input channels num_groups: Number

Map latent-space `[start, end)` coordinates to their pixel-space equivalents by scaling each axis (frame/time, height, width) with the corres

Read video stream metadata as a VideoPixelShape with batch=1. If frame count is missing in the container, decodes the stream to count frames.

Load metadata from path

Create a Modality from a latent state. Constructs a Modality object with the latent state's data, timesteps derived from the denoise mask and

Root-mean-square (RMS) normalize `x` over its last dimension. Thin wrapper around `torch.nn.functional.rms_norm` that infers the normalized sh

Convert the sample and its denoising velocity to denoised sample. Returns: Denoised sample

Create a mapping operation over a set of tiling intervals. The given mapping function is applied to each interval in the input dimension. The resu

Number of tokens after patchification with the default patch size of 1.

Return the guider as-is, or a positive-only guider for absent modalities.

Apply the value operation to the given name and associated value.

Carve per-key tensor views for *layout* into the front of *buffer*. *buffer* is a 1-D ``uint8`` tensor at least :func:`layout_nbytes` long. Each

Decodes video from a file by sequential frame index, without relying on pts. Args: path: Path to the video file. device: Device to

Recursively search for files matching a glob pattern and return the first match.

Run a 3x3x3 Conv3d in-place on a temporally-padded workspace. The workspace has shape ``[B, C, T+2, H, W]`` where positions ``[1:-1]`` hold th

Opens an image file, applies rotation (if it's set in metadata) and converts it to the sRGB color space respecting the original image color s

Signal that an async operation using these weights is guarded by *event*.

Resolve *path* and verify it exists.

Split a dimension into a given number of tiles with overlap. Computes the tile size as ``(dim_size + overlap * (num_tiles - 1)) // num_tiles``

Split a dimension into overlapping tiles of a given size. Tiles are sized ``size`` with ``overlap`` shared elements between consecutive tiles.

Encode video to latent using tiled processing of the given video tensor. Device Handling: - Input video can be on CPU or GPU

Unpatchify the latent state.

Build or update the self-attention mask for newly appended conditioning tokens. If *attention_mask* is ``None`` and no existing mask is present, r

Build caption projections for the transformer when projection is NOT in the text encoder. 19B models: projection is in the transformer (caption_pr

Concatenate a tensor with a padding tensor of the given value.

Send one chat-completions request and return the raw response text.

Clamp a single dim's tile count and overlap to the latent's extent. ``split_by_count`` requires ``overlap < tile_size``; with ``tile_size = (d

Compute float mask for temporal region (prefix or suffix). Returns [B, seq_len] in {0, 1}.

Output `.pt` path relative to the dataset dir; mirrors `process_videos._output_relative` so caption keys match video/audio latent keys (and ab

Encode an audio file through the audio VAE encoder. Returns latent on CPU.

Encode a [B, C, F, H, W] video tensor through the VAE. Returns latent on CPU.