Add custom_vae_version in MBP pre-analysis

Author: atiorh

python_coreml_stable_diffusion/mixed_bit_compression_pre_analysis.py

@@ -570,6 +570,15 @@ def main(args):
         default=7,
         type=int,
     )
+    parser.add_argument(
+        "--custom-vae-version",
+        type=str,
+        default=None,
+        help=
+        ("Custom VAE checkpoint to override the pipeline's built-in VAE. "
+         "If specified, the specified VAE will be converted instead of the one associated to the `--model-version` checkpoint. "
+         "No precision override is applied when using a custom VAE."
+         ))
 
     args = parser.parse_args()
     main(args)