mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
add followlinks to all os.walk() calls
This commit is contained in:
parent
842eb4bb0a
commit
bc16b50302
@ -116,7 +116,7 @@ class MigrateTo3(object):
|
|||||||
appropriate location within the destination models directory.
|
appropriate location within the destination models directory.
|
||||||
"""
|
"""
|
||||||
directories_scanned = set()
|
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:
|
for d in dirs:
|
||||||
try:
|
try:
|
||||||
model = Path(root, d)
|
model = Path(root, d)
|
||||||
|
@ -89,7 +89,7 @@ def get_image_paths(dataroot):
|
|||||||
def _get_paths_from_images(path):
|
def _get_paths_from_images(path):
|
||||||
assert os.path.isdir(path), "{:s} is not a valid directory".format(path)
|
assert os.path.isdir(path), "{:s} is not a valid directory".format(path)
|
||||||
images = []
|
images = []
|
||||||
for dirpath, _, fnames in sorted(os.walk(path)):
|
for dirpath, _, fnames in sorted(os.walk(path, followlinks=True)):
|
||||||
for fname in sorted(fnames):
|
for fname in sorted(fnames):
|
||||||
if is_image_file(fname):
|
if is_image_file(fname):
|
||||||
img_path = os.path.join(dirpath, fname)
|
img_path = os.path.join(dirpath, fname)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user