From 7f695fed397b6ceb27b9393ab0705437616f1f51 Mon Sep 17 00:00:00 2001 From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com> Date: Sun, 12 Feb 2023 00:03:42 +1300 Subject: [PATCH] Ignore safetensor or ckpt files inside diffusers model folders. Basically skips the path if the path has the word diffusers anywhere inside it. --- ldm/invoke/model_manager.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/ldm/invoke/model_manager.py b/ldm/invoke/model_manager.py index 3135931eea..920e2873c6 100644 --- a/ldm/invoke/model_manager.py +++ b/ldm/invoke/model_manager.py @@ -799,15 +799,17 @@ class ModelManager(object): models_folder_safetensors = Path(search_folder).glob("**/*.safetensors") ckpt_files = [x for x in models_folder_ckpt if x.is_file()] - safetensor_files = [x for x in models_folder_safetensors if x.is_file] + safetensor_files = [x for x in models_folder_safetensors if x.is_file()] files = ckpt_files + safetensor_files found_models = [] for file in files: - found_models.append( - {"name": file.stem, "location": str(file.resolve()).replace("\\", "/")} - ) + location = str(file.resolve()).replace("\\", "/") + if 'diffusers' not in location: + found_models.append( + {"name": file.stem, "location": location} + ) return search_folder, found_models