mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
follow symlinks
This commit is contained in:
@@ -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}
|
||||
|
||||
@@ -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
@@ -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)
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user