XiaChunxuan

[BUG]AttributeError: module 'numpy' has no attribute 'product'

3

**Describe the bug** When calling ``` File "~/megatron/core/dist_checkpointing/validation.py", line 460, in _validate_sharding_for_key_flattened [rank0]: or stops[-1] != np.product(local_shape) ``` it will release an error `AttributeError: module 'numpy' has no attribute 'product'`...