-
Notifications
You must be signed in to change notification settings - Fork 42
Open
Description
`WEIGHTS_NAME = "pytorch_model.bin"
WEIGHTS_INDEX_NAME = "pytorch_model.bin.index.json"
def extract_local_archive_file(pretrained_model_name_or_path, subfolder, variant=None):
pretrained_model_name_or_path = str(pretrained_model_name_or_path)
if os.path.isfile(
os.path.join(pretrained_model_name_or_path, subfolder, _add_variant(WEIGHTS_NAME, variant))
):
# Load from a PyTorch checkpoint
archive_file = os.path.join(
pretrained_model_name_or_path, subfolder, _add_variant(WEIGHTS_NAME, variant)
)
return archive_file, False`
utils.py 中 WEIGHTS_NAME 固定写死了只能加载pytorch_model.bin
在模型加载实践中,无法加载 model.safetensors,提示 no file named pytorch_model.bin
请问有什么方法可以支持 safetensors 权重的加载,
目前使用将 safetensors转换成 pytorch_model.bin 仍然加载失败,源码中出现各种参数不存在
Metadata
Metadata
Assignees
Labels
No labels