fix: ensure init-image-strength 0 is respected

addresses https://github.com/brycedrennan/imaginAIry/issues/38
pull/45/head
Bryce 2 years ago committed by Bryce Drennan
parent 302845f35a
commit d689b06256

@ -266,18 +266,24 @@ def imagine(
# encode (scaled latent) # encode (scaled latent)
seed_everything(prompt.seed) seed_everything(prompt.seed)
noise = torch.randn_like(init_latent, device="cpu").to(get_device()) noise = torch.randn_like(init_latent, device="cpu").to(get_device())
z_enc = sampler.stochastic_encode( if generation_strength >= 1:
init_latent, # prompt strength gets converted to time encodings,
torch.tensor([t_enc - 1]).to(get_device()), # which means you can't get to true 0 without this hack
noise=noise, # (or setting steps=1000)
) z_enc = noise
else:
z_enc = sampler.stochastic_encode(
init_latent,
torch.tensor([t_enc - 1]).to(get_device()),
noise=noise,
)
log_latent(z_enc, "z_enc") log_latent(z_enc, "z_enc")
# decode it # decode it
samples = sampler.decode( samples = sampler.decode(
z_enc, x_latent=z_enc,
c, cond=c,
t_enc, t_start=t_enc,
unconditional_guidance_scale=prompt.prompt_strength, unconditional_guidance_scale=prompt.prompt_strength,
unconditional_conditioning=uc, unconditional_conditioning=uc,
img_callback=_img_callback, img_callback=_img_callback,

Loading…
Cancel
Save