filipstrand d80b6e520f Add option to override saved images		2025-03-01 21:34:26 +01:00
..
assets	Dreambooth v1	2024-12-22 19:42:37 +01:00
callbacks	Let before-loop callback accept latents and let StepwiseHandler include initial latent	2025-02-14 05:21:18 +01:00
config	Fix: Handle edge case for img2img	2025-02-14 05:21:35 +01:00
controlnet	Let before-loop callback accept latents and let StepwiseHandler include initial latent	2025-02-14 05:21:18 +01:00
dreambooth	Refactor model config	2025-02-12 22:27:34 +01:00
error	Refactor model config	2025-02-12 22:27:34 +01:00
flux	Let before-loop callback accept latents and let StepwiseHandler include initial latent	2025-02-14 05:21:18 +01:00
latent_creator	Implement general callback mechanism	2025-02-09 15:27:06 +01:00
models	Add PromptCache: Small optimization for repeated generations with same prompt	2025-02-12 22:27:34 +01:00
post_processing	Add option to override saved images	2025-03-01 21:34:26 +01:00
tokenizer	Apply ruff formating (minor manual modifications to some files which are subsequently excluded)	2024-09-21 19:59:12 +02:00
ui	Refactor model config	2025-02-12 22:27:34 +01:00
weights	Allow 3-bit and 6-bit quantization (mlx > v0.21.0)	2025-02-12 22:27:34 +01:00
__init__.py	Refactor model config	2025-02-12 22:27:34 +01:00
generate_controlnet.py	Let before-loop callback accept latents and let StepwiseHandler include initial latent	2025-02-14 05:21:18 +01:00
generate.py	Let before-loop callback accept latents and let StepwiseHandler include initial latent	2025-02-14 05:21:18 +01:00
save.py	Refactor model config	2025-02-12 22:27:34 +01:00
train.py	Dreambooth v1	2024-12-22 19:42:37 +01:00