Fix shard order (#32023)

aa8f86a4 · Brian · GitHub · b3818805 · aa8f86a4
Unverified Commit aa8f86a4 authored Jul 22, 2024 by Brian Committed by GitHub Jul 22, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/llama/convert_llama_weights_to_hf.py src/transformers/models/llama/convert_llama_weights_to_hf.py +1 -1

No files found.
--- a/src/transformers/models/llama/convert_llama_weights_to_hf.py
+++ b/src/transformers/models/llama/convert_llama_weights_to_hf.py
@@ -161,7 +161,7 @@ def write_model(
        # Sharded
        loaded = [
            torch.load(os.path.join(input_base_path, file), map_location="cpu")
-            for file in os.listdir(input_base_path)
+            for file in sorted(os.listdir(input_base_path))
            if file.endswith(".pth")
        ]
    param_count = 0