From 477fb69e033a4b2801d695b11bec0f24f678a51f Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 23 Apr 2023 21:16:46 -0500 Subject: [PATCH] apply lint --- api/onnx_web/diffusers/load.py | 2 +- api/onnx_web/diffusers/utils.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py index 86059ed31..95454ce2b 100644 --- a/api/onnx_web/diffusers/load.py +++ b/api/onnx_web/diffusers/load.py @@ -448,6 +448,6 @@ def patch_pipeline( original_vae = pipe.vae_decoder pipe.vae_decoder = VAEWrapper(server, original_vae) elif hasattr(pipe, "vae"): - pass # TODO: current wrapper does not work with upscaling VAE + pass # TODO: current wrapper does not work with upscaling VAE else: logger.debug("no VAE found to patch") diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py index 4b00370ef..4f82b6cc4 100644 --- a/api/onnx_web/diffusers/utils.py +++ b/api/onnx_web/diffusers/utils.py @@ -129,7 +129,9 @@ def expand_prompt( text_result = self.text_encoder(input_ids=group.astype(np.int32)) logger.trace( - "text encoder produced %s outputs: %s", len(text_result), [t.shape for t in text_result] + "text encoder produced %s outputs: %s", + len(text_result), + [t.shape for t in text_result], ) last_state, _pooled_output, *hidden_states = text_result