Skip to content

Commit 95e3c82

Browse files
committed
correct place for diffusers config search
1 parent dd4fe68 commit 95e3c82

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

optimum/intel/openvino/modeling_diffusion.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -294,8 +294,8 @@ def _save_pretrained(self, save_directory: Union[str, Path]):
294294
dst_path = save_path / OV_XML_FILE_NAME
295295
dst_path.parent.mkdir(parents=True, exist_ok=True)
296296
openvino.save_model(model.model, dst_path, compress_to_fp16=False)
297-
model_dir = model.config.get("_name_or_path", None) or model.model_save_dir
298-
config_path = Path(model_dir) / CONFIG_NAME
297+
model_dir = self.config.get("_name_or_path", None) or model.model_save_dir
298+
config_path = Path(model_dir) / save_path.name / CONFIG_NAME
299299
if config_path.is_file():
300300
config_save_path = save_path / CONFIG_NAME
301301
shutil.copyfile(config_path, config_save_path)

0 commit comments

Comments
 (0)