vkuzometa

/

fp8-opt-125m

like 0

• "torch.FloatStorage",
• "torchao.dtypes.floatx.float8_layout.Float8Layout",
• "torch.BFloat16Storage",
• "torchao.float8.inference.Float8MMConfig",
• "torch.bfloat16",
• "torchao.quantization.quant_api._input_activation_quant_func_fp8",
• "torch._tensor._rebuild_from_type_v2",
• "torch.device",
• "torchao.quantization.linear_activation_quantized_tensor.LinearActivationQuantizedTensor",
• "torch.storage.UntypedStorage",
• "torch._utils._rebuild_tensor_v2",
• "torchao.dtypes.floatx.float8_layout.Float8AQTTensorImpl",
• "torchao.dtypes.affine_quantized_tensor.AffineQuantizedTensor",
• "torchao.quantization.quant_primitives.ZeroPointDomain",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v3",
• "__builtin__.getattr",
• "torch._utils._rebuild_wrapper_subclass",
• "torchao.quantization.granularity.PerRow",
• "torch.serialization._get_layout",
• "torch.float8_e4m3fn"

How to fix it?