Skip to content

Commit

Permalink
Fix bug.
Browse files Browse the repository at this point in the history
  • Loading branch information
lshqqytiger committed May 5, 2024
1 parent a92b859 commit 216340d
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions modules/onnx_impl/pipelines/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,9 @@
import shutil
import tempfile
from typing import Type, Tuple, List, Any, Dict
from packaging import version
import onnx
import torch
import diffusers
import onnxruntime as ort
import optimum.onnxruntime
from modules import shared
from modules.paths import models_path
Expand Down Expand Up @@ -245,7 +244,7 @@ def run_olive(self, submodels: List[str], in_dir: os.PathLike, out_dir: os.PathL
optimized_model_paths = {}

for submodel in submodels:
log.info(f"\nProcessing {submodel}")
print(f"\nProcessing {submodel}")

with open(os.path.join(sd_configs_path, "olive", 'sdxl' if self._is_sdxl else 'sd', f"{submodel}.json"), "r", encoding="utf-8") as config_file:
olive_config: Dict[str, Dict[str, Dict]] = json.load(config_file)
Expand Down Expand Up @@ -282,7 +281,7 @@ def run_olive(self, submodels: List[str], in_dir: os.PathLike, out_dir: os.PathL
**processor_final_pass_footprint["model_config"]["config"]
).model_path

log.info(f"Olive: Successfully processed model: submodel={submodel}")
print(f"Olive: Successfully processed model: submodel={submodel}")

for submodel in submodels:
src_path = optimized_model_paths[submodel]
Expand Down

0 comments on commit 216340d

Please sign in to comment.