Save param shape dtype when use sharding reshard

PaddlePaddle / PaddleNLP

👑 Easy-to-use and powerful NLP and LLM library with 🤗 Awesome model zoo, supporting wide-range of NLP tasks from research to industrial applications, including 🗂Text Classification, 🔍 Neural Search, ❓ Question Answering, ℹ️ Information Extraction, 📄 Document Intelligence, 💌 Sentiment Analysis etc.

https://paddlenlp.readthedocs.io

Apache License 2.0

12k stars 2.93k forks source link

Save param shape dtype when use sharding reshard #8542

Closed sneaxiy closed 3 months ago

sneaxiy commented 3 months ago

PR types

Function optimization

PR changes

Other

Description

Save param shape dtype when use sharding reshard.

paddle-bot[bot] commented 3 months ago

Thanks for your contribution!