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

wip bringing cross-attention to PLMS and DDIM

Browse Source
This commit is contained in:
Damian at mba
2022-10-18 22:09:06 +02:00
parent 09f62032ec
commit 54e6a68acb
6 changed files with 112 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/invoke/generator/txt2img.py
View File

@ -19,7 +19,7 @@ class Txt2Img(Generator):
kwargs are 'width' and 'height'
"""
self.perlin = perlin
uc, c, ec, edit_index_map = conditioning
uc, c, ec, edit_opcodes = conditioning
@torch.no_grad()
def make_image(x_T):
@ -44,7 +44,7 @@ class Txt2Img(Generator):
unconditional_guidance_scale = cfg_scale,
unconditional_conditioning = uc,
edited_conditioning = ec,
edit_token_index_map = edit_index_map,
conditioning_edit_opcodes = edit_opcodes,
eta = ddim_eta,
img_callback = step_callback,
threshold = threshold,