Skip to content

Commit

Permalink
apply lint
Browse files Browse the repository at this point in the history
ssube committed Apr 24, 2023

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
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.