Update on "[reland][quant][eagermode] Move custom_module registration…

… to prepare/convert_custom_config_dict (#46293)"

Summary:

Test Plan: Imported from OSS

Reviewed By: raghuramank100

Differential Revision: [D24322747](https://our.internmc.facebook.com/intern/diff/D24322747)

[ghstack-poisoned]

torch/quantization/quantize.py

@@ -192,14 +192,16 @@ def prepare(model, inplace=False, allow_list=None,
         `prepare_custom_config_dict`: customization configuration dictionary for prepare function
 
     Example:
-    prepare_custom_config_dict = {
-    # user will manually define the corresponding observed
-    # module class which has a from_float class method that converts
-    # float custom module to observed custom module
-      "float_to_observed_custom_module_class": {
-         CustomModule: ObservedCustomModule
-      }
-    }
+
+    .. testcode::
+        prepare_custom_config_dict = {
+            # user will manually define the corresponding observed
+            # module class which has a from_float class method that converts
+            # float custom module to observed custom module
+            "float_to_observed_custom_module_class": {
+                CustomModule: ObservedCustomModule
+            }
+         }
 
     """
     torch._C._log_api_usage_once("quantization_api.quantize.prepare")
@@ -410,14 +412,15 @@ def convert(
         `convert_custom_config_dict`: custom configuration dictionary for convert function
 
     Example:
-    convert_custom_config_dict = {
-    # user will manually define the corresponding quantized
-    # module class which has a from_observed class method that converts
-    # observed custom module to quantized custom module
-      "observed_to_quantized_custom_module_class": {
-         ObservedCustomModule: QuantizedCustomModule
-      }
-    }
+    .. testcode::
+        convert_custom_config_dict = {
+            # user will manually define the corresponding quantized
+            # module class which has a from_observed class method that converts
+            # observed custom module to quantized custom module
+            "observed_to_quantized_custom_module_class": {
+                ObservedCustomModule: QuantizedCustomModule
+            }
+         }
 
     """
     torch._C._log_api_usage_once("quantization_api.quantize.convert")