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 hi-res fix

Browse Source
This commit is contained in:
Damian at mba 2022-10-18 22:54:51 +02:00
parent d572af2acf
commit 2b79a716aa
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ldm/invoke/generator/txt2img2img.py
View File

@ -22,7 +22,7 @@ class Txt2Img2Img(Generator):
Return value depends on the seed at the time you call it
kwargs are 'width' and 'height'
"""
uc, c = conditioning
uc, c, ec, edit_opcodes = conditioning
@torch.no_grad()
def make_image(x_T):
@ -60,7 +60,9 @@ class Txt2Img2Img(Generator):
unconditional_guidance_scale = cfg_scale,
unconditional_conditioning = uc,
eta = ddim_eta,
img_callback = step_callback
img_callback = step_callback,
edited_conditioning = ec,
conditioning_edit_opcodes = edit_opcodes
)
print(
@ -94,6 +96,8 @@ class Txt2Img2Img(Generator):
img_callback = step_callback,
unconditional_guidance_scale=cfg_scale,
unconditional_conditioning=uc,
edited_conditioning = ec,
conditioning_edit_opcodes = edit_opcodes
)
if self.free_gpu_mem:

3
ldm/models/diffusion/sampler.py
View File

@ -309,6 +309,9 @@ class Sampler(object):
use_original_steps=False,
init_latent = None,
mask = None,
edited_conditioning = None,
conditioning_edit_opcodes = None,
**kwargs
):
timesteps = (