add is_distributed_field in sharding reshard (#8875)

Author: sneaxiy

paddlenlp/trainer/utils/sharding_io.py

@@ -588,7 +588,8 @@ def _gather_sharding_metas(self):
         param_meta = {}
         for k, v in model.state_dict().items():
             structure_name_mapping[k] = v.name
-            param_meta[k] = (v.shape, int(v.dtype))
+            is_distributed = getattr(v, "is_distributed", False)
+            param_meta[k] = (v.shape, int(v.dtype), is_distributed)
 
         sharding_metas = {}
         sharding_meta = {}