mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
cleanup
This commit is contained in:
@@ -97,7 +97,6 @@ class DiffusionSampler:
|
||||
if 'predict_x0' in self.config:
|
||||
self.config['predict_x0'] = opts.uni_pc_variant
|
||||
if name == 'DPM++ 2M':
|
||||
print('HERE', opts.schedulers_dpm_solver)
|
||||
self.config['algorithm_type'] = opts.schedulers_dpm_solver
|
||||
if 'beta_start' in self.config and opts.schedulers_beta_start > 0:
|
||||
self.config['beta_start'] = opts.schedulers_beta_start
|
||||
|
||||
@@ -262,6 +262,7 @@ class EmbeddingDatabase:
|
||||
self.previously_displayed_embeddings = displayed_embeddings
|
||||
shared.log.info(f"Loaded embeddings: loaded={len(self.word_embeddings)} skipped={len(self.skipped_embeddings)}")
|
||||
|
||||
|
||||
def find_embedding_at_position(self, tokens, offset):
|
||||
token = tokens[offset]
|
||||
possible_matches = self.ids_lookup.get(token, None)
|
||||
|
||||
@@ -152,7 +152,9 @@ def create_output_panel(tabname, outdir):
|
||||
import modules.generation_parameters_copypaste as parameters_copypaste
|
||||
|
||||
def open_folder(gallery):
|
||||
if gallery is not None and len(gallery) > 0:
|
||||
if type(gallery) is str:
|
||||
folder = gallery
|
||||
elif gallery is not None and len(gallery) > 0:
|
||||
folder = os.path.dirname(gallery[-1]['name'])
|
||||
else:
|
||||
folder = shared.opts.outdir_samples or outdir
|
||||
|
||||
@@ -19,19 +19,18 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
|
||||
sd_models.model_data.sd_model.embedding_db.load_textual_inversion_embeddings(force_reload=True)
|
||||
|
||||
def list_items(self):
|
||||
def list_folder(folder):
|
||||
for filename in os.listdir(folder):
|
||||
fn = os.path.join(folder, filename)
|
||||
if os.path.isfile(fn) and (fn.lower().endswith(".pt") or fn.lower().endswith(".safetensors")):
|
||||
embedding = Embedding(vec=0, name=os.path.basename(fn), filename=fn)
|
||||
embedding.filename = fn
|
||||
embeddings.append(embedding)
|
||||
elif os.path.isdir(fn) and not fn.startswith('.'):
|
||||
list_folder(fn)
|
||||
|
||||
if sd_models.model_data.sd_model is None:
|
||||
embeddings = []
|
||||
|
||||
def list_folder(folder):
|
||||
for filename in os.listdir(folder):
|
||||
fn = os.path.join(folder, filename)
|
||||
if os.path.isfile(fn) and (fn.lower().endswith(".pt") or fn.lower().endswith(".safetensors")):
|
||||
embedding = Embedding(vec=0, name=os.path.basename(fn), filename=fn)
|
||||
embedding.filename = fn
|
||||
embeddings.append(embedding)
|
||||
elif os.path.isdir(fn) and not fn.startswith('.'):
|
||||
list_folder(fn)
|
||||
|
||||
list_folder(shared.opts.embeddings_dir)
|
||||
elif shared.backend == shared.Backend.ORIGINAL:
|
||||
embeddings = list(sd_hijack.model_hijack.embedding_db.word_embeddings.values())
|
||||
|
||||
Reference in New Issue
Block a user