Unverified Commit 8d48d0a9 authored by Balaxxe's avatar Balaxxe Committed by GitHub
Browse files

[Bugfix] Sort hf_weights_files in fastsafetensors_weights_iterator to match #33491 (#34190)


Signed-off-by: default avatarBalaxxe <136368465+jaim12005@users.noreply.github.com>
parent 9608844f
......@@ -826,6 +826,7 @@ def fastsafetensors_weights_iterator(
pg = SingleGroup()
device = torch.device(f"cuda:{current_platform.current_device()}")
hf_weights_files = sorted(hf_weights_files, key=_natural_sort_key)
weight_files_sub_lists = [
hf_weights_files[i : i + pg.size()]
for i in range(0, len(hf_weights_files), pg.size())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment