From 5023cac7be2ea42bfa2ec4acd2e576a2598c1ee3 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Tue, 10 Sep 2024 16:00:36 +0200 Subject: [PATCH] update model parts to use frozen dict --- optimum/onnxruntime/modeling_diffusion.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/optimum/onnxruntime/modeling_diffusion.py b/optimum/onnxruntime/modeling_diffusion.py index 18cd38c5f2..131625da78 100644 --- a/optimum/onnxruntime/modeling_diffusion.py +++ b/optimum/onnxruntime/modeling_diffusion.py @@ -39,6 +39,8 @@ StableDiffusionXLImg2ImgPipeline, StableDiffusionXLPipeline, ) +from diffusers.configuration_utils import FrozenDict +from diffusers.image_processor import VaeImageProcessor from diffusers.schedulers.scheduling_utils import SCHEDULER_CONFIG_NAME from diffusers.utils import CONFIG_NAME, is_invisible_watermark_available from huggingface_hub import snapshot_download @@ -58,7 +60,6 @@ from ..pipelines.diffusers.pipeline_stable_diffusion_inpaint import StableDiffusionInpaintPipelineMixin from ..pipelines.diffusers.pipeline_stable_diffusion_xl import StableDiffusionXLPipelineMixin from ..pipelines.diffusers.pipeline_stable_diffusion_xl_img2img import StableDiffusionXLImg2ImgPipelineMixin -from ..pipelines.diffusers.pipeline_utils import VaeImageProcessor from ..utils import ( DIFFUSION_MODEL_TEXT_ENCODER_2_SUBFOLDER, DIFFUSION_MODEL_TEXT_ENCODER_SUBFOLDER, @@ -503,10 +504,9 @@ def __init__(self, session: ort.InferenceSession, parent_model: ORTPipeline): config_path = Path(session._model_path).parent / self.CONFIG_NAME if config_path.is_file(): - # TODO: use FrozenDict - self.config = parent_model._dict_from_json_file(config_path) + self.config = FrozenDict(parent_model._dict_from_json_file(config_path)) else: - self.config = {} + self.config = FrozenDict({}) super().__init__(session, parent_model)