diffusers

/

FLUX.1-dev-torchao-fp8

like 1

Follow

🧨Diffusers 525

• "torch.float8_e4m3fn",
• "torch._utils._rebuild_wrapper_subclass",
• "torch._tensor._rebuild_from_type_v2",
• "torch.BFloat16Storage",
• "collections.OrderedDict",
• "torchao.dtypes.floatx.float8_layout.Float8Layout",
• "torch._utils._rebuild_tensor_v2",
• "torch.serialization._get_layout",
• "torchao.dtypes.affine_quantized_tensor.AffineQuantizedTensor",
• "torchao.dtypes.floatx.float8_layout.Float8AQTTensorImpl",
• "torch.device",
• "torch.storage.UntypedStorage",
• "torch._utils._rebuild_tensor_v3",
• "torchao.quantization.quant_primitives.ZeroPointDomain",
• "torch.bfloat16"

How to fix it?

• "torch.bfloat16",
• "torch._tensor._rebuild_from_type_v2",
• "torchao.quantization.quant_primitives.ZeroPointDomain",
• "torch._utils._rebuild_tensor_v2",
• "torchao.dtypes.floatx.float8_layout.Float8Layout",
• "torch.serialization._get_layout",
• "torchao.dtypes.affine_quantized_tensor.AffineQuantizedTensor",
• "torch.device",
• "torch.BFloat16Storage",
• "collections.OrderedDict",
• "torchao.dtypes.floatx.float8_layout.Float8AQTTensorImpl",
• "torch._utils._rebuild_tensor_v3",
• "torch._utils._rebuild_wrapper_subclass",
• "torch.storage.UntypedStorage",
• "torch.float8_e4m3fn"

How to fix it?