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