How to speed up the method of obtaining checkpoint types.

[suzukimain]

I am trying to get the checkpoint type as follows, but when I try to get the model type in ckpt format, it is very slow and also occupies a lot of memory.

import time

from huggingface_hub import hf_hub_download
from diffusers.loaders.single_file_utils import (
    load_single_file_checkpoint,
    infer_diffusers_model_type
    )

safetensors_path = hf_hub_download(repo_id="stable-diffusion-v1-5/stable-diffusion-v1-5", filename="v1-5-pruned-emaonly.safetensors")
ckpt_path = hf_hub_download(repo_id="stable-diffusion-v1-5/stable-diffusion-v1-5", filename="v1-5-pruned-emaonly.ckpt")

safetensors_start_time = time.time()
safetensors_model_type = infer_diffusers_model_type(load_single_file_checkpoint(safetensors_path))
safetensors_end_time = time.time()

start_time = time.time()
ckpt_type = infer_diffusers_model_type(load_single_file_checkpoint(ckpt_path))
end_time = time.time()

print(f"safetensors_model_type: {safetensors_model_type}, time: {safetensors_end_time - safetensors_start_time}")
print(f"ckpt_model_type: {ckpt_type}, time: {end_time - start_time}")

Perhaps the torch.load() part called within load_single_file_checkpoint is the cause, could you please advise?

diffusers/src/diffusers/models/model_loading_utils.py

Lines 145 to 149 in 1dbd26f

	return torch.load(
	checkpoint_file,
	map_location="cpu",
	**weights_only_kwarg,
	)

[suzukimain]

Can anyone give me some advice?