[diffusion] quant: add support for svdquant and nunchaku (#18549)

Co-authored-by: AichenF <aichenf@nvidia.com>
Co-authored-by: jianyingzhu <53300651@qq.com>
This commit is contained in:
Mick
2026-02-15 20:43:00 +08:00
committed by GitHub
co-authored by AichenF jianyingzhu
parent 88010e9601
commit 3feb48139e
20 changed files with 1304 additions and 123 deletions
@@ -86,6 +86,8 @@ def hf_to_custom_state_dict(
target_param_name, merge_index, num_params_to_merge = param_names_mapping(
source_param_name
)
if target_param_name == "" or target_param_name is None: # type: ignore[comparison-overlap]
continue
reverse_param_names_mapping[target_param_name] = (
source_param_name,
merge_index,