let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: https://github.com/Sanster/iopaint-docs/issues/47

Browse Source
This commit is contained in:
Qing
2024-11-14 17:56:28 +08:00
parent 668733c178
commit 14b334ad76
1 changed files with 42 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
iopaint/download.py
View File

@@ -61,7 +61,7 @@ def get_sd_model_type(model_abs_path: str) -> Optional[ModelType]:
model_abs_path, model_abs_path,
load_safety_checker=False, load_safety_checker=False,
num_in_channels=9, num_in_channels=9,
original_config_file=get_config_files()['v1'] original_config_file=get_config_files()["v1"],
) )
model_type = ModelType.DIFFUSERS_SD_INPAINT model_type = ModelType.DIFFUSERS_SD_INPAINT
except ValueError as e: except ValueError as e:
@@ -89,7 +89,7 @@ def get_sdxl_model_type(model_abs_path: str) -> Optional[ModelType]:
model_abs_path, model_abs_path,
load_safety_checker=False, load_safety_checker=False,
num_in_channels=9, num_in_channels=9,
original_config_file=get_config_files()['xl'], original_config_file=get_config_files()["xl"],
) )
if model.unet.config.in_channels == 9: if model.unet.config.in_channels == 9:
# https://github.com/huggingface/diffusers/issues/6610 # https://github.com/huggingface/diffusers/issues/6610
@@ -207,11 +207,13 @@ def scan_diffusers_models() -> List[ModelInfo]:
cache_dir = Path(HF_HUB_CACHE) cache_dir = Path(HF_HUB_CACHE)
# logger.info(f"Scanning diffusers models in {cache_dir}") # logger.info(f"Scanning diffusers models in {cache_dir}")
diffusers_model_names = [] diffusers_model_names = []
model_index_files = glob.glob(os.path.join(cache_dir, "**/*", "model_index.json"), recursive=True) model_index_files = glob.glob(
os.path.join(cache_dir, "**/*", "model_index.json"), recursive=True
)
for it in model_index_files: for it in model_index_files:
it = Path(it) it = Path(it)
with open(it, "r", encoding="utf-8") as f:
try: try:
with open(it, "r", encoding="utf-8") as f:
data = json.load(f) data = json.load(f)
except: except:
continue continue
@@ -255,7 +257,9 @@ def _scan_converted_diffusers_models(cache_dir) -> List[ModelInfo]:
cache_dir = Path(cache_dir) cache_dir = Path(cache_dir)
available_models = [] available_models = []
diffusers_model_names = [] diffusers_model_names = []
model_index_files = glob.glob(os.path.join(cache_dir, "**/*", "model_index.json"), recursive=True) model_index_files = glob.glob(
os.path.join(cache_dir, "**/*", "model_index.json"), recursive=True
)
for it in model_index_files: for it in model_index_files:
it = Path(it) it = Path(it)
with open(it, "r", encoding="utf-8") as f: with open(it, "r", encoding="utf-8") as f: