Ryan Dick
49ca42f84a
Initial (untested) implementation of MultiDiffusionPipeline.
2024-06-18 15:35:52 -04:00
Ryan Dick
493fcd8660
Remove inpainting support from MultiDiffusionPipeline.
2024-06-18 15:35:52 -04:00
Ryan Dick
20322d781e
Remove IP-Adapter and T2I-Adapter support from MultiDiffusionPipeline.
2024-06-18 15:35:52 -04:00
Ryan Dick
889d13e02a
Document plan for the rest of the MultiDiffusion implementation.
2024-06-18 15:35:52 -04:00
Ryan Dick
6ccd2a867b
Add detailed docstring to latents_from_embeddings().
2024-06-18 15:35:52 -04:00
Ryan Dick
5861fa1719
Copy StableDiffusionGeneratorPipeline as a starting point for a new MultiDiffusionPipeline.
2024-06-18 15:35:52 -04:00
Ryan Dick
dfd4beb62b
Simplify handling of inpainting models. Improve the in-code documentation around inpainting.
2024-06-18 15:35:52 -04:00
Ryan Dick
83df0c0df5
Minor tidying of latents_from_embeddings(...).
2024-06-18 15:35:52 -04:00
Ryan Dick
c58c4069a7
Consolidate latents_from_embeddings(...) and generate_latents_from_embeddings(...) into a single function.
2024-06-18 15:35:52 -04:00
Ryan Dick
3937fffa94
Fix invocation name of tiled_multi_diffusion_denoise_latents.
2024-06-18 15:35:52 -04:00
Ryan Dick
bbf5f67691
Improve clarity of comments regarded when 'noise' and 'latents' are expected to be set.
2024-06-18 15:35:52 -04:00
Ryan Dick
2f5c147b84
Fix static check errors on imports in diffusers_pipeline.py.
2024-06-18 15:35:52 -04:00
Ryan Dick
bd2839b748
Remove a condition for handling inpainting models that never resolves to True. The same logic is already applied earlier by AddsMaskLatents.
2024-06-18 15:35:52 -04:00
Ryan Dick
4f70dd7ce1
Add clarifying comment to explain why noise might be None in latents_from_embedding().
2024-06-18 15:35:52 -04:00
Ryan Dick
066672fbfd
Remove unused are_like_tensors() function.
2024-06-18 15:35:52 -04:00
Ryan Dick
abefaee4d1
Remove unused StableDiffusionGeneratorPipeline.use_ip_adapter member.
2024-06-18 15:35:52 -04:00
Ryan Dick
3254ba5904
Remove unused StableDiffusionGeneratorPipeline.control_model.
2024-06-18 15:35:52 -04:00
Ryan Dick
73a8c55852
Stricter typing for the is_gradient_mask: bool.
2024-06-18 15:35:52 -04:00
Ryan Dick
f82af7c22d
Fix typing of control_data to reflect that it can be None.
2024-06-18 15:35:52 -04:00
Ryan Dick
3aef717ef4
Fix typing of timesteps and init_timestep.
2024-06-18 15:35:52 -04:00
Ryan Dick
c2cf1137e9
Fix typing to reflect that the callback arg to latents_from_embeddings is never None.
2024-06-18 15:35:52 -04:00
Ryan Dick
803a24bc0a
Move seed above optional params.
2024-06-18 15:35:52 -04:00
Ryan Dick
7d24ad8ccd
Simplify handling of AddsMaskGuidance, and fix some related type errors.
2024-06-18 15:35:52 -04:00
Ryan Dick
cb389063b2
Remove unused num_inference_steps.
2024-06-18 15:35:52 -04:00
Ryan Dick
81b8a69e1a
WIP TiledMultiDiffusionDenoiseLatents. Updated parameter list and first half of the logic.
2024-06-18 15:35:50 -04:00
Ryan Dick
7ee5db87ad
Tidy DenoiseLatentsInvocation.prep_control_data(...) and fix some type errors.
2024-06-18 15:34:30 -04:00
Ryan Dick
66cf2c59bd
Make DenoiseLatentsInvocation.prep_control_data(...) a staticmethod so that it can be called externally.
2024-06-18 15:34:30 -04:00
Ryan Dick
3bad1367e9
Copy TiledStableDiffusionRefineInvocation as a starting point for TiledMultiDiffusionDenoiseLatents.py
2024-06-18 15:34:22 -04:00
Ryan Dick
867a7642a6
Change tiling strategy to make TiledStableDiffusionRefineInvocation work with more tile shapes and overlaps.
2024-06-18 15:31:58 -04:00
Ryan Dick
d9d1c8f9cb
Expose a few more params from TiledStableDiffusionRefineInvocation.
2024-06-18 15:31:58 -04:00
Ryan Dick
e03eb7fb45
Add support for LoRA models in TiledStableDiffusionRefineInvocation.
2024-06-18 15:31:58 -04:00
Ryan Dick
85db33bc7e
Add naive ControlNet support to TiledStableDiffusionRefineInvocation
2024-06-18 15:31:58 -04:00
Ryan Dick
93e3a2b504
Fix ControlNetModel type hint import source.
2024-06-18 15:31:58 -04:00
Ryan Dick
6a7a26f1bf
Rough prototype of TiledStableDiffusionRefineInvocation is working.
2024-06-18 15:31:58 -04:00
Ryan Dick
08ca03ef9f
WIP - TiledStableDiffusionRefine
2024-06-18 15:31:54 -04:00
Ryan Dick
ccf90b6bd6
Minor improvements to LatentsToImageInvocation type hints.
2024-06-18 15:31:21 -04:00
Ryan Dick
753239b48d
Expose vae_decode(...) as a staticmethod on LatentsToImageInvocation.
2024-06-18 15:31:21 -04:00
Ryan Dick
65fa4664c9
Fix return type of prepare_noise_and_latents(...).
2024-06-18 15:31:21 -04:00
Ryan Dick
297570ded3
Make init_scheduler() a staticmethod on DenoiseLatentsInvocation so that it can be called externally.
2024-06-18 15:31:21 -04:00
Ryan Dick
680fdcf293
Only allow a single positive/negative prompt conditioning input for tiled refine.
2024-06-18 15:31:21 -04:00
Ryan Dick
5ff91f2c44
WIP on TiledStableDiffusionRefine
2024-06-18 15:31:14 -04:00
Ryan Dick
69aa7057e7
Convert several methods in DenoiseLatentsInvocation to staticmethods so that they can be called externally.
2024-06-18 15:25:08 -04:00
Ryan Dick
d3932f40de
Simplify the logic in prepare_noise_and_latents(...).
2024-06-18 15:25:08 -04:00
Ryan Dick
ee74cd7fab
Split out the prepare_noise_and_latents(...) logic in DenoiseLatentsInvocation so that it can be called from other invocations.
2024-06-18 15:25:08 -04:00
Ryan Dick
bda25b40c9
(minor) Add a TODO note to get_scheduler(...).
2024-06-18 15:25:08 -04:00
Ryan Dick
79ceac2f82
(minor) Use SilenceWarnings as a decorator rather than a context manager to save an indentation level.
2024-06-18 15:06:22 -04:00
Ryan Dick
8e47e005a7
Tidy SilenceWarnings context manager:
...
- Fix type errors
- Enable SilenceWarnings to be used as both a context manager and a decorator
- Remove duplicate implementation
- Check the initial verbosity on __enter__() rather than __init__()
2024-06-18 15:06:22 -04:00
Ryan Dick
d13aafb514
Tidy denoise_latents.py imports to all use absolute import paths.
2024-06-18 15:06:22 -04:00
Brandon Rising
63a7e19dbf
Run ruff
2024-06-18 10:38:29 -04:00
Brandon Rising
fbc5a8ec65
Ignore validation on improperly formatted hashes (pytest)
2024-06-18 10:38:29 -04:00
