Skip to content

Commit 7809a1a

Browse files
committed
doc fix
Signed-off-by: ynankani <ynankani@nvidia.com>
1 parent dd84179 commit 7809a1a

1 file changed

Lines changed: 7 additions & 8 deletions

File tree

modelopt/torch/export/unified_export_hf.py

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1111,14 +1111,13 @@ def export_hf_checkpoint(
11111111
components: Only used for diffusers pipelines. Optional list of component names
11121112
to export. If None, all quantized components are exported.
11131113
extra_state_dict: Extra state dictionary to add to the exported model.
1114-
**kwargs: Internal-only keyword arguments. Supported keys:
1115-
merged_base_safetensor_path (str | None): If provided, merge the exported
1116-
diffusion transformer weights with non-transformer components (VAE, vocoder,
1117-
text encoders, etc.) from this base safetensors file and add quantization
1118-
metadata to produce a single-file checkpoint compatible with ComfyUI. This
1119-
should be the path to a full base model ``.safetensors`` file,
1120-
e.g. ``"path/to/ltx-2-19b-dev.safetensors"``.
1121-
Only used for diffusion model exports.
1114+
**kwargs: Internal-only keyword arguments. Supported key: merged_base_safetensor_path
1115+
(str, optional). When provided, merges the exported diffusion transformer
1116+
weights with non-transformer components (VAE, vocoder, text encoders, etc.)
1117+
from this base safetensors file to produce a single-file checkpoint
1118+
compatible with ComfyUI. Value should be the path to a full base model
1119+
``.safetensors`` file (e.g. ``"path/to/ltx-2-19b-dev.safetensors"``).
1120+
Only used for diffusion model exports.
11221121
"""
11231122
merged_base_safetensor_path: str | None = kwargs.get("merged_base_safetensor_path")
11241123
export_dir = Path(export_dir)

0 commit comments

Comments
 (0)