fix `load_state_dict` for xpu and refine xpu safetensor version check

huggingface / accelerate

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

https://huggingface.co/docs/accelerate

Apache License 2.0

7.32k stars 872 forks source link

fix `load_state_dict` for xpu and refine xpu safetensor version check #2879

Open faaany opened 1 week ago

faaany commented 1 week ago

What does this PR do?

enable test_load_state_dict on xpu, because torch.device(0) by default is cuda
add xpu support in load_state_dict when device_map's values are identical integers
refine the xpu safetensor version check to make code clean and compact

faaany commented 1 week ago

@SunMarc this PR is ready for review. Could you help review it? Thx a lot!

HuggingFaceDocBuilderDev commented 32 minutes ago

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.