unsloth

/

FLUX.2-dev-FP8

like 1

Follow

Unsloth AI 29.8k

• "torch._utils._rebuild_wrapper_subclass",
• "torch._utils._rebuild_tensor_v3",
• "torch.storage.UntypedStorage",
• "torch.FloatStorage",
• "torchao.float8.inference.Float8MMConfig",
• "torch.float8_e4m3fn",
• "torchao.quantization.Float8Tensor",
• "collections.OrderedDict",
• "torch.BFloat16Storage",
• "torchao.quantization.quantize_.common.kernel_preference.KernelPreference",
• "torch.bfloat16",
• "torchao.quantization.granularity.PerRow",
• "torch._tensor._rebuild_from_type_v2",
• "torch._utils._rebuild_tensor_v2",
• "torchao.quantization.quantize_.workflows.float8.float8_tensor.QuantizeTensorToFloat8Kwargs",
• "torch.torch_version.TorchVersion",
• "torch.device",
• "torch.serialization._get_layout"

How to fix it?

• "torch.CharStorage",
• "torch.device",
• "torchao.dtypes.utils.PlainLayout",
• "torch.bfloat16",
• "torchao.dtypes.affine_quantized_tensor.AffineQuantizedTensor",
• "torch.int8",
• "torch._utils._rebuild_wrapper_subclass",
• "torch.BFloat16Storage",
• "torch.serialization._get_layout",
• "torchao.quantization.quant_primitives.ZeroPointDomain",
• "torchao.quantization.quant_api._int8_symm_per_token_reduced_range_quant",
• "torchao.quantization.linear_activation_quantized_tensor.LinearActivationQuantizedTensor",
• "torchao.dtypes.uintx.plain_layout.PlainAQTTensorImpl",
• "torch.torch_version.TorchVersion",
• "collections.OrderedDict",
• "torch._tensor._rebuild_from_type_v2",
• "torch._utils._rebuild_tensor_v2"

How to fix it?