sayakpaul

/

flux.1-schell-int8wo

like 2

• "torch._utils._rebuild_tensor_v2",
• "torch.LongStorage",
• "collections.OrderedDict",
• "torchao.dtypes.affine_quantized_tensor.AffineQuantizedTensor",
• "torch._utils._rebuild_wrapper_subclass",
• "torch.int8",
• "torch.BFloat16Storage",
• "torch.serialization._get_layout",
• "torchao.quantization.quant_primitives.ZeroPointDomain",
• "torchao.dtypes.utils.PlainLayoutType",
• "torch.device",
• "torch._tensor._rebuild_from_type_v2",
• "torchao.dtypes.affine_quantized_tensor.PlainAQTLayout",
• "torch.bfloat16",
• "torch.CharStorage"

How to fix it?