Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 08:44:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Vladimir Mandic
2023-05-20 13:14:04 -04:00
parent f8f81f86e6
commit 2ce9852cff
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/prompt_parser.py
+5 -3
View File
@@ -1,9 +1,11 @@
# pylint: disable=anomalous-backslash-in-string
"""
import os
import sys
from rich import print
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
"""
import re
from collections import namedtuple
@@ -367,9 +369,9 @@ def parse_prompt_attention(text):
if __name__ == "__main__":
input_text = '[black] [[grey]] (white) ((gray)) ((orange:1.1) yellow) ((purple) and [dark] red:1.1) [mouse:0.2] [(cat:1.1):0.5]'
print(f'Prompt: {input_text}')
schedules = get_learned_conditioning_prompt_schedules([input_text], 100)[0]
print('Schedules', schedules)
for schedule in schedules:
all_schedules = get_learned_conditioning_prompt_schedules([input_text], 100)[0]
print('Schedules', all_schedules)
for schedule in all_schedules:
print('Schedule', schedule[0])
opts.data['prompt_attention'] = 'Fixed attention'
output_list = parse_prompt_attention(schedule[1])