import re cn_model_module = { "inpaint": "inpaint_global_harmonious", "scribble": "t2ia_sketch_pidi", "lineart": "lineart_coarse", "openpose": "openpose_full", "tile": "tile_resample", "depth": "depth_midas", } _names = [*cn_model_module, "union"] cn_model_regex = re.compile("|".join(_names), flags=re.IGNORECASE)