deal with non-directories in diffusers/

2024-08-30 20:32:17 +00:00 · 2023-03-05 12:29:52 -05:00 · 2023-03-05 12:29:52 -05:00 · fc187f263e
commit fc187f263e
parent 4e9e1b660d
1 changed files with 4 additions and 1 deletions
--- a/invokeai/backend/model_management/model_manager.py
+++ b/invokeai/backend/model_management/model_manager.py
@ -966,7 +966,10 @@ class ModelManager(object):
                if dest.is_symlink():
                    print(f"** Found symlink at {dest.name}. Not migrating.")
                elif dest.exists():
-                    rmtree(source)
+                    if source.is_dir():
+                        rmtree(source)
+                    else:
+                        source.unlink()
                else:
                    move(source, dest)