We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f7b5b54 commit 03a59aaCopy full SHA for 03a59aa
optimum/intel/openvino/modeling_diffusion.py
@@ -741,6 +741,7 @@ def __init__(
741
self.model_name = model_name
742
self.parent_pipeline = parent_pipeline
743
self.request = None if not parent_pipeline._compile_only else self.model
744
+ self.ov_config = parent_pipeline.ov_config
745
746
if isinstance(parent_pipeline.model_save_dir, TemporaryDirectory):
747
self.model_save_dir = Path(parent_pipeline.model_save_dir.name) / self.model_name
@@ -764,10 +765,6 @@ def _device(self) -> str:
764
765
def device(self) -> torch.device:
766
return self.parent_pipeline.device
767
- @property
768
- def ov_config(self) -> OVConfig:
769
- return self.parent_pipeline.ov_config
770
-
771
@property
772
def dtype(self) -> torch.dtype:
773
return OV_TO_PT_TYPE[self.ov_config.get("dtype", "f32")]
0 commit comments