Skip to content

Commit 52af45a

Browse files
committed
Fixing info prints to be more consistent.
1 parent bac8438 commit 52af45a

File tree

2 files changed

+1
-7
lines changed

2 files changed

+1
-7
lines changed

gimpopenvino/plugins/openvino_utils/tools/openvino_common/models_ov/stable_diffusion_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def __init__(self, model="runwayml/stable-diffusion-v1-5",
120120
if "NPU" in device:
121121
try_enable_npu_turbo(device, self.core)
122122

123-
print("Loading models... ")
123+
print("Loading models ... int8 ")
124124

125125

126126

testscases/StableDiffusion/stable_diffusion_engine_tc.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -123,27 +123,21 @@ def print_system_info():
123123

124124
def initialize_engine(model_name, model_path, device_list):
125125
if model_name == "sd_1.5_square_int8":
126-
log.info('Device list: %s', device_list)
127126
return stable_diffusion_engine.StableDiffusionEngineAdvanced(model=model_path, device=device_list)
128127
if model_name == "sd_3.0_square":
129128
device_list = ["GPU"]
130-
log.info('Device list: %s', device_list)
131129
return stable_diffusion_3.StableDiffusionThreeEngine(model=model_path, device=device_list)
132130
if model_name == "sd_1.5_inpainting":
133131
return stable_diffusion_engine_inpainting_genai.StableDiffusionEngineInpaintingGenai(model=model_path, device=device_list[0])
134132
if model_name in ("sd_1.5_square_lcm","sdxl_base_1.0_square","sdxl_turbo_square","sd_3.0_med_diffuser_square","sd_3.5_med_turbo_square"):
135133
return stable_diffusion_engine_genai.StableDiffusionEngineGenai(model=model_path,model_name=model_name,device=device_list)
136134
if model_name == "sd_1.5_inpainting_int8":
137-
log.info('Advanced Inpainting Device list: %s', device_list)
138135
return stable_diffusion_engine_inpainting_advanced.StableDiffusionEngineInpaintingAdvanced(model=model_path, device=device_list)
139136
if model_name == "controlnet_openpose_int8":
140-
log.info('Device list: %s', device_list)
141137
return controlnet_openpose_advanced.ControlNetOpenPoseAdvanced(model=model_path, device=device_list)
142138
if model_name == "controlnet_canny_int8":
143-
log.info('Device list: %s', device_list)
144139
return controlnet_cannyedge_advanced.ControlNetCannyEdgeAdvanced(model=model_path, device=device_list)
145140
if model_name == "controlnet_scribble_int8":
146-
log.info('Device list: %s', device_list)
147141
return controlnet_scribble.ControlNetScribbleAdvanced(model=model_path, device=device_list)
148142
if model_name == "controlnet_canny":
149143
return controlnet_canny_edge.ControlNetCannyEdge(model=model_path, device=device_list)

0 commit comments

Comments
 (0)