Skip to content

Commit 03a59aa

Browse files
authored
updated OVPipelinePart to have separate ov_config (#957)
1 parent f7b5b54 commit 03a59aa

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

optimum/intel/openvino/modeling_diffusion.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -741,6 +741,7 @@ def __init__(
741741
self.model_name = model_name
742742
self.parent_pipeline = parent_pipeline
743743
self.request = None if not parent_pipeline._compile_only else self.model
744+
self.ov_config = parent_pipeline.ov_config
744745

745746
if isinstance(parent_pipeline.model_save_dir, TemporaryDirectory):
746747
self.model_save_dir = Path(parent_pipeline.model_save_dir.name) / self.model_name
@@ -764,10 +765,6 @@ def _device(self) -> str:
764765
def device(self) -> torch.device:
765766
return self.parent_pipeline.device
766767

767-
@property
768-
def ov_config(self) -> OVConfig:
769-
return self.parent_pipeline.ov_config
770-
771768
@property
772769
def dtype(self) -> torch.dtype:
773770
return OV_TO_PT_TYPE[self.ov_config.get("dtype", "f32")]

0 commit comments

Comments
 (0)