Skip to content

Commit

Permalink
apply lint
Browse files Browse the repository at this point in the history
ssube committed Apr 24, 2023
1 parent b1aafd2 commit 477fb69
Showing 2 changed files with 4 additions and 2 deletions.
2 changes: 1 addition & 1 deletion api/onnx_web/diffusers/load.py
Original file line number Diff line number Diff line change
@@ -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")
4 changes: 3 additions & 1 deletion api/onnx_web/diffusers/utils.py
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit 477fb69

Please sign in to comment.