Ryan Dick
|
bd74b84cc5
|
Revert "Remove the redundant init_timestep parameter that was being passed around. It is simply the first element of the timesteps array."
This reverts commit fa40061eca.
|
2024-06-26 12:51:51 -04:00 |
|
Ryan Dick
|
7c032ea604
|
(minor) Fix some documentation typos.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
fa40061eca
|
Remove the redundant init_timestep parameter that was being passed around. It is simply the first element of the timesteps array.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
ceae5fe1db
|
(minor) typo
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
60ac937698
|
Improve clarity of comments regarded when 'noise' and 'latents' are expected to be set.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
196f3b721d
|
Stricter typing for the is_gradient_mask: bool.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
40ae174c41
|
Fix typing of timesteps and init_timestep.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
ffc28176fe
|
Remove unused num_inference_steps.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
7e94350351
|
Tidy DenoiseLatentsInvocation.prep_control_data(...) and fix some type errors.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
c4e8549c73
|
Make DenoiseLatentsInvocation.prep_control_data(...) a staticmethod so that it can be called externally.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
020e8eb413
|
Fix return type of prepare_noise_and_latents(...).
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
3d49541c09
|
Make init_scheduler() a staticmethod on DenoiseLatentsInvocation so that it can be called externally.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
171a505f5e
|
Convert several methods in DenoiseLatentsInvocation to staticmethods so that they can be called externally.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
8004a0d5f5
|
Simplify the logic in prepare_noise_and_latents(...).
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
610a1fd611
|
Split out the prepare_noise_and_latents(...) logic in DenoiseLatentsInvocation so that it can be called from other invocations.
|
2024-06-25 11:31:52 -07:00 |
|
Ryan Dick
|
43108eec13
|
(minor) Add a TODO note to get_scheduler(...).
|
2024-06-25 11:31:52 -07: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
|
d13aafb514
|
Tidy denoise_latents.py imports to all use absolute import paths.
|
2024-06-18 15:06:22 -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 |
|
Ryan Dick
|
52c0c4a32f
|
Rename latent.py -> denoise_latents.py.
|
2024-06-07 09:28:42 -04:00 |
|