diff --git a/backends/apple/coreml/README.md b/backends/apple/coreml/README.md index b3b22ed9997..e8a062774d0 100644 --- a/backends/apple/coreml/README.md +++ b/backends/apple/coreml/README.md @@ -93,14 +93,14 @@ class Model(torch.nn.Module): source_model = Model() example_inputs = (torch.randn((1, 3, 256, 256)), ) -pre_autograd_aten_dialect = export_for_training(model, example_inputs).module() +pre_autograd_aten_dialect = export_for_training(source_model, example_inputs).module() quantization_config = LinearQuantizerConfig.from_dict( { "global_config": { "quantization_scheme": QuantizationScheme.symmetric, - "activation_dtype": torch.uint8, - "weight_dtype": torch.int8, + "activation_dtype": torch.quint8, + "weight_dtype": torch.qint8, "weight_per_channel": True, } }