img2img works with all samplers, inpainting working with ddim & plms

- img2img confirmed working with all samplers
- inpainting working on ddim & plms. Changes to k-diffusion
  module seem to be needed for inpainting support.
- switched k-diffuser noise schedule to original karras schedule,
  which reduces the step number needed for good results
This commit is contained in:
Lincoln Stein
2022-09-25 04:03:28 -04:00
parent 72834ad16c
commit 958d7650dd
11 changed files with 597 additions and 685 deletions

View File

@ -324,6 +324,7 @@ def main_loop(gen, opt, infile):
opt.last_operation='generate'
gen.prompt2image(
image_callback=image_writer,
# step_callback=gen.write_intermediate_images(5,'./outputs/img-samples/intermediates'), #DEBUGGING ONLY - DELETE
catch_interrupts=catch_ctrl_c,
**vars(opt)
)