Skip to content

Commit 5e27fc7

Browse files
committed
Restore compatibility with old ONNX pipeline.
I think it broke in #552.
1 parent 813744e commit 5e27fc7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -259,6 +259,7 @@ def __call__(
259259
class StableDiffusionOnnxPipeline(OnnxStableDiffusionPipeline):
260260
def __init__(
261261
self,
262+
vae_encoder: OnnxRuntimeModel,
262263
vae_decoder: OnnxRuntimeModel,
263264
text_encoder: OnnxRuntimeModel,
264265
tokenizer: CLIPTokenizer,
@@ -270,6 +271,7 @@ def __init__(
270271
deprecation_message = "Please use `OnnxStableDiffusionPipeline` instead of `StableDiffusionOnnxPipeline`."
271272
deprecate("StableDiffusionOnnxPipeline", "1.0.0", deprecation_message)
272273
super().__init__(
274+
vae_encoder=vae_encoder,
273275
vae_decoder=vae_decoder,
274276
text_encoder=text_encoder,
275277
tokenizer=tokenizer,

0 commit comments

Comments
 (0)