Erland

/

vanilla-340M-4096-model-AO-W4

like 0

• "torchao.quantization.quant_primitives.ZeroPointDomain",
• "torchao.dtypes.affine_quantized_tensor.AffineQuantizedTensor",
• "torchao.dtypes.uintx.tensor_core_tiled_layout.TensorCoreTiledAQTTensorImpl",
• "torch.serialization._get_layout",
• "collections.OrderedDict",
• "torch.IntStorage",
• "torch.int32",
• "torch._tensor._rebuild_from_type_v2",
• "torch._utils._rebuild_tensor_v2",
• "torch.bfloat16",
• "torch.BFloat16Storage",
• "torchao.dtypes.uintx.tensor_core_tiled_layout.TensorCoreTiledLayout",
• "torch._utils._rebuild_wrapper_subclass",
• "torch.device"

How to fix it?