Upload mp_rank_02_001
155ac63 distrib_optim.pt Detected Pickle imports (4)
- "collections.OrderedDict",
- "torch.float32",
- "torch.FloatStorage",
- "torch._utils._rebuild_tensor_v2"
How to fix it?
11.5 GB Upload mp_rank_02_001 model_optim_rng.pt Detected Pickle imports (12)
- "megatron.model.enums.PositionEmbeddingType",
- "torch._utils._rebuild_tensor_v2",
- "numpy.core.multiarray._reconstruct",
- "torch.BFloat16Storage",
- "torch.ByteStorage",
- "numpy.dtype",
- "numpy.ndarray",
- "megatron.core.enums.ModelType",
- "collections.OrderedDict",
- "torch.bfloat16",
- "_codecs.encode",
- "argparse.Namespace"
How to fix it?
1.92 GB Upload mp_rank_02_001
