diff --git a/invokeai/backend/install/migrate_to_3.py b/invokeai/backend/install/migrate_to_3.py index 3f360f291d..0938ed2199 100644 --- a/invokeai/backend/install/migrate_to_3.py +++ b/invokeai/backend/install/migrate_to_3.py @@ -116,7 +116,7 @@ class MigrateTo3(object): appropriate location within the destination models directory. """ directories_scanned = set() - for root, dirs, files in os.walk(src_dir): + for root, dirs, files in os.walk(src_dir, followlinks=True): for d in dirs: try: model = Path(root, d) diff --git a/invokeai/backend/stable_diffusion/image_degradation/utils_image.py b/invokeai/backend/stable_diffusion/image_degradation/utils_image.py index d45ca602e6..fb71f85e5a 100644 --- a/invokeai/backend/stable_diffusion/image_degradation/utils_image.py +++ b/invokeai/backend/stable_diffusion/image_degradation/utils_image.py @@ -89,7 +89,7 @@ def get_image_paths(dataroot): def _get_paths_from_images(path): assert os.path.isdir(path), "{:s} is not a valid directory".format(path) images = [] - for dirpath, _, fnames in sorted(os.walk(path)): + for dirpath, _, fnames in sorted(os.walk(path, followlinks=True)): for fname in sorted(fnames): if is_image_file(fname): img_path = os.path.join(dirpath, fname)