follow symlinks

This commit is contained in:
Vladimir Mandic
2023-05-18 07:36:38 -04:00
parent 4d67ee67ed
commit c1e70df845
5 changed files with 8 additions and 8 deletions
+1 -3
View File
@@ -1,9 +1,7 @@
import glob
import os
import re
import torch
from typing import Union
import torch
from modules import shared, devices, sd_models, errors, scripts
metadata_tags_order = {"ss_sd_model_name": 1, "ss_resolution": 2, "ss_clip_skip": 3, "ss_num_train_images": 10, "ss_tag_frequency": 20}
+4 -2
View File
@@ -186,8 +186,10 @@ div#extras_scale_to_tab div.form{
}
.extra-network-cards{
height: 725px;
overflow: scroll;
height: fit-content;
max-height: 50vh;
overflow-y: scroll;
overflow-x: hidden;
resize: vertical;
}
+1 -1
View File
@@ -733,7 +733,7 @@ def walk_files(path, allowed_extensions=None):
return
if allowed_extensions is not None:
allowed_extensions = set(allowed_extensions)
for root, _dirs, files in os.walk(path):
for root, _dirs, files in os.walk(path, followlinks=True):
for filename in files:
if allowed_extensions is not None:
_, ext = os.path.splitext(filename)
+1 -1
View File
@@ -81,7 +81,7 @@ class ExtraNetworksPage:
subdirs = {}
allowed_folders = [os.path.abspath(x) for x in self.allowed_directories_for_previews()]
for parentdir in [*set(allowed_folders)]:
for root, dirs, _files in os.walk(parentdir):
for root, dirs, _files in os.walk(parentdir, followlinks=True):
for dirname in dirs:
x = os.path.join(root, dirname)
if not os.path.isdir(x):
@@ -16,7 +16,7 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
def list_items(self):
embeddings = [emb for emb in sd_hijack.model_hijack.embedding_db.word_embeddings.values()]
if len(embeddings) == 0: # maybe not loaded yet, so lets just look them up
for root, _dirs, fns in os.walk(shared.opts.embeddings_dir):
for root, _dirs, fns in os.walk(shared.opts.embeddings_dir, followlinks=True):
for fn in fns:
if fn.lower().endswith(".pt"):
embedding = Embedding(0, fn)