Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

onnx python 3.9 compatability

Browse Source
This commit is contained in:
Seunghoon Lee
2024-02-01 19:01:11 +09:00
parent 016fd03a38
commit 90b6052872
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/onnx_impl/pipelines.py
+4 -4
View File
@@ -1045,9 +1045,9 @@ class OnnxStableDiffusionXLPipeline(PipelineBase, optimum.onnxruntime.ORTStableD
vae_encoder: Optional[ort.InferenceSession] = None,
text_encoder_2: Optional[ort.InferenceSession] = None,
tokenizer_2: Any = None,
use_io_binding: bool | None = None,
use_io_binding: Optional[bool] = None,
model_save_dir = None,
add_watermarker: bool | None = None
add_watermarker: Optional[bool] = None
):
super(optimum.onnxruntime.ORTStableDiffusionXLPipeline, self).__init__(vae_decoder, text_encoder, unet, config, tokenizer, scheduler, feature_extractor, vae_encoder, text_encoder_2, tokenizer_2, use_io_binding, model_save_dir, add_watermarker)
@@ -1071,9 +1071,9 @@ class OnnxStableDiffusionXLImg2ImgPipeline(PipelineBase, optimum.onnxruntime.ORT
vae_encoder: Optional[ort.InferenceSession] = None,
text_encoder_2: Optional[ort.InferenceSession] = None,
tokenizer_2: Any = None,
use_io_binding: bool | None = None,
use_io_binding: Optional[bool] = None,
model_save_dir = None,
add_watermarker: bool | None = None
add_watermarker: Optional[bool] = None
):
super(optimum.onnxruntime.ORTStableDiffusionXLImg2ImgPipeline, self).__init__(vae_decoder, text_encoder, unet, config, tokenizer, scheduler, feature_extractor, vae_encoder, text_encoder_2, tokenizer_2, use_io_binding, model_save_dir, add_watermarker)