ministral3 (#14251)

Signed-off-by: Xinyuan Tong <xinyuantong.cs@gmail.com>
Co-authored-by: Yueming Yuan <yy28@illinois.edu>
This commit is contained in:
Xinyuan Tong
2025-12-04 22:31:26 +00:00
committed by GitHub
parent c1006fd8a1
commit 6d37e70883
14 changed files with 245 additions and 26 deletions

View File

@@ -529,6 +529,14 @@ def filter_duplicate_safetensors_files(
# torch state_dict to safetensors file holding that weight.
index_file_name = os.path.join(hf_folder, index_file)
if not os.path.isfile(index_file_name):
# NOTE: this is a trick of handling mistral model
# skip the unsupported consolidated.safetensors file
if len(hf_weights_files) == 2:
hf_weights_files.sort()
if hf_weights_files[0].endswith(
"consolidated.safetensors"
) and hf_weights_files[1].endswith("model.safetensors"):
return [hf_weights_files[1]]
return hf_weights_files
# Iterate through the weight_map (weight_name: safetensors files)