TorchAO Testing
Upload OPTForCausalLM
ae95421 verified - 1.52 kB initial commit
- 5.17 kB Upload OPTForCausalLM
- 1.96 kB Upload OPTForCausalLM
- 132 Bytes Upload OPTForCausalLM
pytorch_model.bin Detected Pickle imports (15)
- "torch.serialization._get_layout",
- "torch.storage.UntypedStorage",
- "torch._utils._rebuild_tensor_v2",
- "torch.bfloat16",
- "torchao.quantization.Float8Tensor",
- "torch._utils._rebuild_tensor_v3",
- "torch._utils._rebuild_wrapper_subclass",
- "torch.device",
- "torch.BFloat16Storage",
- "torch.FloatStorage",
- "torchao.quantization.granularity.PerRow",
- "collections.OrderedDict",
- "torch.float8_e4m3fn",
- "torch._tensor._rebuild_from_type_v2",
- "torchao.float8.inference.Float8MMConfig"
How to fix it?
166 MB Upload OPTForCausalLM