Skip to content

Commit

Permalink
upgrade optimum.
Browse files Browse the repository at this point in the history
  • Loading branch information
@lshqqytiger
lshqqytiger committed Oct 27, 2024
1 parent 7c94652 commit 96eebe8
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions modules/onnx_impl/__init__.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -189,8 +189,8 @@ def preprocess_pipeline(p):
return shared.sd_model


def ORTDiffusionModelPart_to(self, *args, **kwargs):
self.parent_model = self.parent_model.to(*args, **kwargs)
def ORTDiffusionModelPart_to(self: optimum.onnxruntime.modeling_diffusion.ORTPipelinePart, *args, **kwargs):
self.parent_pipeline = self.parent_pipeline.to(*args, **kwargs)
return self


Expand Down Expand Up @@ -241,7 +241,7 @@ def initialize_onnx():
diffusers.ORTStableDiffusionXLPipeline = diffusers.OnnxStableDiffusionXLPipeline # Huggingface model compatibility
diffusers.ORTStableDiffusionXLImg2ImgPipeline = diffusers.OnnxStableDiffusionXLImg2ImgPipeline

optimum.onnxruntime.modeling_diffusion._ORTDiffusionModelPart.to = ORTDiffusionModelPart_to # pylint: disable=protected-access
optimum.onnxruntime.modeling_diffusion.ORTPipelinePart.to = ORTDiffusionModelPart_to # pylint: disable=protected-access

fastapi_encoders.jsonable_encoder = jsonable_encoder

Expand Down

1 comment on commit 96eebe8

@Roninos
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What is it?
launch.py 48
main()

launch.py 44 main
start()

launch_utils.py 712 start
import webui

webui.py 13
initialize.imports()

initialize.py 35 imports
from modules import shared_init

shared_init.py 8
from modules.zluda import initialize_zluda

zluda.py 7
from modules.onnx_impl.execution_providers import available_execution_providers, ExecutionProvider

init.py 192
def ORTDiffusionModelPart_to(self: optimum.onnxruntime.modeling_diffusion.ORTPipelinePart, *args, **kwargs):

AttributeError:
module 'optimum.onnxruntime.modeling_diffusion' has no attribute 'ORTPipelinePart'

Please sign in to comment.