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