File tree Expand file tree Collapse file tree 1 file changed +1
-4
lines changed Expand file tree Collapse file tree 1 file changed +1
-4
lines changed Original file line number Diff line number Diff 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" )]
You can’t perform that action at this time.
0 commit comments