@@ -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