Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

timesteps with hires

Browse Source
This commit is contained in:
Vladimir Mandic
2024-05-12 17:38:33 -04:00
parent 3757d0c780
commit c20e080b8a
3 changed files with 22 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/styles.py
+15 -9
View File
@@ -80,19 +80,25 @@ def apply_wildcards_to_prompt(prompt, all_wildcards):
if len(prompt) == 0:
return prompt
replaced = {}
t0 = time.time()
for style_wildcards in all_wildcards:
wildcards = [x.strip() for x in style_wildcards.replace('\n', ' ').split(";") if len(x.strip()) > 0]
for wildcard in wildcards:
what, words = wildcard.split("=", 1)
words = [x.strip() for x in words.split(",") if len(x.strip()) > 0]
word = random.choice(words)
prompt = prompt.replace(what, word)
replaced[what] = word
t0 = time.time()
prompt, replaced, not_found = apply_file_wildcards(prompt, [], [])
try:
what, words = wildcard.split("=", 1)
words = [x.strip() for x in words.split(",") if len(x.strip()) > 0]
word = random.choice(words)
prompt = prompt.replace(what, word)
replaced[what] = word
except Exception as e:
shared.log.error(f'Wildcards: wildcard="{wildcard}" error={e}')
t1 = time.time()
if len(replaced) > 0 or len(not_found) > 0:
shared.log.info(f'Wildcards applied: {replaced} missing: {not_found} path="{shared.opts.wildcards_dir}" time={t1-t0:.2f}')
prompt, replaced_file, not_found = apply_file_wildcards(prompt, [], [])
t2 = time.time()
if replaced:
shared.log.info(f'Wildcards applied: {replaced} path="{shared.opts.wildcards_dir}" type=style time={t1-t0:.2f}')
if len(replaced_file) > 0 or len(not_found) > 0:
shared.log.info(f'Wildcards applied: {replaced_file} missing: {not_found} path="{shared.opts.wildcards_dir}" type=file time={t2-t2:.2f} ')
return prompt