filipstrand e3b2a65aa6 Allow 3-bit and 6-bit quantization (mlx > v0.21.0)		2025-02-12 22:27:34 +01:00
..
assets	Dreambooth v1	2024-12-22 19:42:37 +01:00
callbacks	Implement general callback mechanism	2025-02-09 15:27:06 +01:00
config	Refactor: Remove ConfigControlnet	2025-02-09 14:37:23 +01:00
controlnet	Add PromptCache: Small optimization for repeated generations with same prompt	2025-02-12 22:27:34 +01:00
dreambooth	Refactor: Transformer & Flux	2025-02-02 19:12:07 +01:00
error	Dreambooth v1	2024-12-22 19:42:37 +01:00
flux	Add PromptCache: Small optimization for repeated generations with same prompt	2025-02-12 22:27:34 +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	Implement general callback mechanism	2025-02-09 15:27:06 +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	Allow 3-bit and 6-bit quantization (mlx > v0.21.0)	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: Remove ConfigControlnet	2025-02-09 14:37:23 +01:00
generate_controlnet.py	Implement general callback mechanism	2025-02-09 15:27:06 +01:00
generate.py	Implement general callback mechanism	2025-02-09 15:27:06 +01:00
save.py	comprehensive ModelConfig refactor to support compatible HuggingFace dev/schnell models	2025-01-17 06:27:33 -08:00
train.py	Dreambooth v1	2024-12-22 19:42:37 +01:00