TorchAO Testing
Update README.md
a5b10a9 verified - 1.52 kB initial commit
- 1.29 kB Update README.md
model.pt Detected Pickle imports (20)
- "torch._tensor._rebuild_from_type_v2",
- "torchao.quantization.granularity.PerRow",
- "torch.bfloat16",
- "torchao.quantization.linear_activation_quantized_tensor.LinearActivationQuantizedTensor",
- "torchao.dtypes.floatx.float8_layout.Float8AQTTensorImpl",
- "torch.float8_e4m3fn",
- "torch._utils._rebuild_wrapper_subclass",
- "collections.OrderedDict",
- "torch.storage.UntypedStorage",
- "torch._utils._rebuild_tensor_v3",
- "torch.FloatStorage",
- "torchao.quantization.quant_api._input_activation_quant_func_fp8",
- "torch.serialization._get_layout",
- "torchao.dtypes.floatx.float8_layout.Float8Layout",
- "torch.device",
- "torch.BFloat16Storage",
- "torchao.dtypes.affine_quantized_tensor.AffineQuantizedTensor",
- "torchao.float8.inference.Float8MMConfig",
- "torch._utils._rebuild_tensor_v2",
- "torchao.quantization.quant_primitives.ZeroPointDomain"
How to fix it?
13.1 kB Rename model.bin to model.pt - 1.71 kB Upload model_inputs.pt with huggingface_hub
- 2.6 kB Upload model_output.pt with huggingface_hub