diffusers
b695febc - fix(torchao): route root-level tensors around safetensors reconstruction

Commit
9 days ago
fix(torchao): route root-level tensors around safetensors reconstruction Models with parameters at the root of the module tree (e.g. Wan's `scale_shift_table`) crashed torchao's `unflatten_tensor_state_dict` when loading serialized checkpoints, since flattened tensor names are assumed to carry a module prefix. Filter such tensors and their metadata entries out of the reconstruction and merge them back unchanged. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Author
sayak@huggingface.co
Committer
sayak@huggingface.co
Parents
Loading