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

cross attention control options

Browse Source
This commit is contained in:
Damian at mba
2022-10-23 14:58:25 +02:00
parent 8273c04575
commit 7d677a63b8
12 changed files with 318 additions and 299 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ldm/generate.py
View File

@ -400,7 +400,7 @@ class Generate:
mask_image = None
try:
uc, c, ec, ec_index_map = get_uc_and_c_and_ec(
uc, c, extra_conditioning_info = get_uc_and_c_and_ec(
prompt, model =self.model,
skip_normalize=skip_normalize,
log_tokens =self.log_tokenization
@ -438,7 +438,7 @@ class Generate:
sampler=self.sampler,
steps=steps,
cfg_scale=cfg_scale,
conditioning=(uc, c, ec, ec_index_map),
conditioning=(uc, c, extra_conditioning_info),
ddim_eta=ddim_eta,
image_callback=image_callback, # called after the final image is generated
step_callback=step_callback, # called after each intermediate image is generated
@ -541,8 +541,8 @@ class Generate:
image = Image.open(image_path)
# used by multiple postfixers
# todo: cross-attention
uc, c, _, _ = get_uc_and_c_and_ec(
# todo: cross-attention control
uc, c, _ = get_uc_and_c_and_ec(
prompt, model =self.model,
skip_normalize=opt.skip_normalize,
log_tokens =opt.log_tokenization