fix error (#279)

Author: jiangdi0924

python_coreml_stable_diffusion/mixed_bit_compression_apply.py

@@ -136,6 +136,15 @@ def op_selector(const):
         type=str,
         help=("The string key into --pre-analysis-json-path's baselines dict"
     ))
+    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()
 

python_coreml_stable_diffusion/mixed_bit_compression_pre_analysis.py

@@ -18,6 +18,7 @@
 
 import torch
 import torch.nn as nn
+import requests
 torch.set_grad_enabled(False)
 
 from tqdm import tqdm, trange