diff --git a/api/onnx_web/chain/blend_controlnet.py b/api/onnx_web/chain/blend_controlnet.py
index f3f430d98..f8efa9a49 100644
--- a/api/onnx_web/chain/blend_controlnet.py
+++ b/api/onnx_web/chain/blend_controlnet.py
@@ -35,7 +35,7 @@ def blend_controlnet(
         params.model,
         params.scheduler,
         job.get_device(),
-        control=None,   # TODO: needs to have a ControlNet
+        control=None,  # TODO: needs to have a ControlNet
     )
 
     rng = np.random.RandomState(params.seed)
diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
index 2331c9ca2..46d8ff3b8 100644
--- a/api/onnx_web/diffusers/load.py
+++ b/api/onnx_web/diffusers/load.py
@@ -310,7 +310,6 @@ def load_pipeline(
             model,
             provider=device.ort_provider(),
             sess_options=device.sess_options(),
-            revision="onnx",
             safety_checker=None,
             torch_dtype=torch_dtype,
             **components,
diff --git a/api/onnx_web/diffusers/pipelines/pix2pix.py b/api/onnx_web/diffusers/pipelines/pix2pix.py
index 180dc0076..492f224fb 100644
--- a/api/onnx_web/diffusers/pipelines/pix2pix.py
+++ b/api/onnx_web/diffusers/pipelines/pix2pix.py
@@ -425,7 +425,7 @@ def __call__(
                 ):
                     progress_bar.update()
                     if callback is not None and i % callback_steps == 0:
-                        callback(i, t, latents.numpy())
+                        callback(i, t, latents)
 
         # 10. Post-processing
         image = self.decode_latents(latents)
diff --git a/api/onnx_web/utils.py b/api/onnx_web/utils.py
index 49b844f42..1e1f875be 100644
--- a/api/onnx_web/utils.py
+++ b/api/onnx_web/utils.py
@@ -38,7 +38,9 @@ def get_and_clamp_int(
     return min(max(int(args.get(key, default_value)), min_value), max_value)
 
 
-def get_from_list(args: Any, key: str, values: Sequence[Any], default_value: Optional[Any] = None) -> Optional[Any]:
+def get_from_list(
+    args: Any, key: str, values: Sequence[Any], default_value: Optional[Any] = None
+) -> Optional[Any]:
     selected = args.get(key, default_value)
     if selected in values:
         return selected