@@ -123,27 +123,21 @@ def print_system_info():
123123
124124def 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