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

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

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