From 11e422cf297c4e30a880afab5ed770e7dd81d1b8 Mon Sep 17 00:00:00 2001 From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com> Date: Sun, 12 Feb 2023 00:13:22 +1300 Subject: [PATCH] Ignore two files names instead of the entire folder rather than bypassing any path with diffusers in it, im specifically bypassing model.safetensors and diffusion_pytorch_model.safetensors both of which should be diffusers files in most cases. --- ldm/invoke/model_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/invoke/model_manager.py b/ldm/invoke/model_manager.py index 920e2873c6..0e7a0747ab 100644 --- a/ldm/invoke/model_manager.py +++ b/ldm/invoke/model_manager.py @@ -806,7 +806,7 @@ class ModelManager(object): found_models = [] for file in files: location = str(file.resolve()).replace("\\", "/") - if 'diffusers' not in location: + if 'model.safetensors' not in location and 'diffusion_pytorch_model.safetensors' not in location: found_models.append( {"name": file.stem, "location": location} )