Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

make concepts library work with Web UI

Browse Source 
This PR makes it possible to include a Hugging Face concepts library
<style-or-subject-trigger> in the WebUI prompt. The metadata seems
to be correctly handled.
This commit is contained in:
Lincoln Stein 2022-11-28 23:37:46 +00:00
parent 45e51bac9a
commit 62cda009dd
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ldm/generate.py
View File

@ -436,6 +436,9 @@ class Generate:
self.sampler_name = sampler_name self.sampler_name = sampler_name
self._set_sampler() self._set_sampler()
# apply the concepts library to the prompt
prompt = self.concept_lib().replace_concepts_with_triggers(prompt, lambda concepts: self.load_concepts(concepts))
# bit of a hack to change the cached sampler's karras threshold to # bit of a hack to change the cached sampler's karras threshold to
# whatever the user asked for # whatever the user asked for
if karras_max is not None and isinstance(self.sampler,KSampler): if karras_max is not None and isinstance(self.sampler,KSampler):

2
ldm/invoke/CLI.py
View File

@ -352,8 +352,6 @@ def main_loop(gen, opt):
last_results.append([path, seed]) last_results.append([path, seed])
if operation == 'generate': if operation == 'generate':
# load any <embeddings> from the SD concepts library
opt.prompt = gen.concept_lib().replace_concepts_with_triggers(opt.prompt, lambda concepts: gen.load_concepts(concepts))
catch_ctrl_c = infile is None # if running interactively, we catch keyboard interrupts catch_ctrl_c = infile is None # if running interactively, we catch keyboard interrupts
opt.last_operation='generate' opt.last_operation='generate'
try: try: