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

enable --hires to use k* samplers

Browse Source
This commit is contained in:
Lincoln Stein 2022-10-05 17:18:32 -04:00
parent bcb6e2e506
commit 2f29b78a00
1 changed files with 2 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ldm/dream/generator/txt2img2img.py
View File

@ -78,26 +78,14 @@ class Txt2Img2Img(Generator):
x = self.get_noise(width,height,False) x = self.get_noise(width,height,False)
# Other samplers not supported yet, so ignore previous sampler z_enc = sampler.stochastic_encode(
if not isinstance(sampler,DDIMSampler):
print(
f"\n>> Sampler '{sampler.__class__.__name__}' is not yet supported for img2img. Using DDIM sampler"
)
img_sampler = DDIMSampler(self.model, device=self.model.device)
img_sampler.make_schedule(
ddim_num_steps=steps, ddim_eta=ddim_eta, verbose=False
)
else:
img_sampler = sampler
z_enc = img_sampler.stochastic_encode(
samples, samples,
torch.tensor([t_enc]).to(self.model.device), torch.tensor([t_enc]).to(self.model.device),
noise=x noise=x
) )
# decode it # decode it
samples = img_sampler.decode( samples = sampler.decode(
z_enc, z_enc,
c, c,
t_enc, t_enc,