Skip to content

Commit 1ca606a

Browse files
committed
Added status updates for Model manager UI
1 parent 76ebaed commit 1ca606a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

gimpopenvino/plugins/openvino_utils/tools/model_manager.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -920,8 +920,10 @@ def _download_model(self, model_id):
920920
json.dump(config, json_file, indent=4)
921921

922922
if model_id == "sd_15_inpainting":
923+
self.model_install_status[model_id]["status"] = "Generating IRs..."
923924
export_command = f"{Path(optimum_ex)} export openvino --model {Path(full_download_folder)} --weight-format fp16 --task image-to-image {Path(full_install_path)}"
924925
else:
926+
self.model_install_status[model_id]["status"] = "Generating IRs..."
925927
export_command = f"{Path(optimum_ex)} export openvino --model {Path(full_download_folder)} --weight-format fp16 --task stable-diffusion {Path(full_install_path)}"
926928
print("Running the command:", export_command)
927929

@@ -937,6 +939,7 @@ def _download_model(self, model_id):
937939
if "sdxl_base" in model_id:
938940
model_name="sdxl_base_1.0_square"
939941
if "sdxl" in model_id:
942+
self.model_install_status[model_id]["status"] = "Model Caching..."
940943
stable_diffusion_engine_genai.StableDiffusionEngineGenai(model=full_install_path,model_name=model_name,device=["GPU","GPU","GPU"])
941944
if config["power modes supported"] == "yes":
942945
stable_diffusion_engine_genai.StableDiffusionEngineGenai(model=full_install_path,model_name=model_name,device=["GPU","NPU","GPU"])

0 commit comments

Comments
 (0)