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
Brandon Rising
8ce6e4540e
Run ruff
2024-06-18 10:38:29 -04:00
Brandon Rising
f14f377ede
Update validator list
2024-06-18 10:38:29 -04:00
Brandon Rising
1925f83f5e
Update validator list
2024-06-18 10:38:29 -04:00
Brandon Rising
3a5ad6d112
Update validator list
2024-06-18 10:38:29 -04:00
Brandon Rising
41a6bb45f3
Initial functionality
2024-06-18 10:38:29 -04:00
psychedelicious
cd70937b7f
feat(api): improved model install confirmation page styling & messaging
2024-06-17 10:51:08 +10:00
psychedelicious
f002bca2fa
feat(ui): handle new model_install_download_started event
...
When a model install is initiated from outside the client, we now trigger the model manager tab's model install list to update.
- Handle new `model_install_download_started` event
- Handle `model_install_download_complete` event (this event is not new but was never handled)
- Update optimistic updates/cache invalidation logic to efficiently update the model install list
2024-06-17 10:07:10 +10:00
psychedelicious
56771de856
feat(ui): add redux actions for model_install_download_started event
2024-06-17 09:52:46 +10:00
psychedelicious
c11478a94a
chore(ui): typegen
2024-06-17 09:51:18 +10:00
psychedelicious
fb694b3e17
feat(app): add model_install_download_started event
...
Previously, we used `model_install_download_progress` for both download starting and progressing. When handling this event, we don't know which actual thing it represents.
Add `model_install_download_started` event to explicitly represent a model download started event.
2024-06-17 09:50:25 +10:00
psychedelicious
1bc98abc76
docs(ui): explain model install events
2024-06-17 09:33:46 +10:00
chainchompa
7f03b04b2f
Merge branch 'main' into chainchompa/model-install-deeplink
2024-06-14 17:16:25 -04:00
chainchompa
4029972530
formatting
2024-06-14 17:15:55 -04:00
chainchompa
328f160e88
refetch model installs when a new model install starts
2024-06-14 17:09:07 -04:00
chainchompa
aae318425d
added route for installing huggingface model from model marketplace
2024-06-14 17:08:39 -04:00
Ryan Dick
785bb1d9e4
Fix all comparisons against the DEFAULT_PRECISION constant. DEFAULT_PRECISION is a torch.dtype. Previously, it was compared to a str in a number of places where it would always resolve to False. This is a bugfix that results in a change to the default behavior. In practice, this will not change the behavior for many users, because it only causes a change in behavior if a users has configured float32 as their default precision.
2024-06-14 11:26:10 -07:00
