torchao-dev

/

opt-125m-FP8-v2-0.13-dev

like 0

• "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?