Billy
|
4377158503
|
Variant
|
2025-03-13 13:32:57 +11:00 |
|
Billy
|
d8b9a8d0dd
|
Merge branch 'main' into model-classification-api
|
2025-03-13 13:03:51 +11:00 |
|
Billy
|
39a4608d15
|
Fix annotations compatability 3.11
|
2025-03-13 13:01:19 +11:00 |
|
Billy
|
b86ac5e049
|
Explicit union
|
2025-03-13 10:28:07 +11:00 |
|
Billy
|
665236bb79
|
Type hints
|
2025-03-13 09:21:58 +11:00 |
|
Billy
|
f45400a275
|
Remove hash algo
|
2025-03-12 18:39:29 +11:00 |
|
psychedelicious
|
e35537e60a
|
fix(mm): move flux_redux starter model to the flux bundle, make siglip a dependency of it
|
2025-03-11 11:17:19 +11:00 |
|
Billy
|
d86b392bfd
|
Remove redundant hash_algo field
|
2025-03-11 09:16:59 +11:00 |
|
Billy
|
3e9e45b177
|
Update comments
|
2025-03-11 09:04:19 +11:00 |
|
Billy
|
907d960745
|
PR suggestions
|
2025-03-11 08:37:43 +11:00 |
|
Billy
|
bfdace6437
|
New API for model classification
|
2025-03-11 08:34:34 +11:00 |
|
psychedelicious
|
cf0cbaf0ae
|
chore: ruff (more)
|
2025-03-06 10:57:54 +11:00 |
|
psychedelicious
|
ac6fc6eccb
|
chore: ruff
|
2025-03-06 10:57:54 +11:00 |
|
Ryan Dick
|
8e28888bc4
|
Fix SigLipPipeline model size calculation.
|
2025-03-06 10:31:17 +11:00 |
|
Ryan Dick
|
4c02ba908a
|
Add support for FLUX Redux masks.
|
2025-03-06 10:31:17 +11:00 |
|
Ryan Dick
|
f1fde792ee
|
Get FLUX Redux working: model loading and inference.
|
2025-03-06 10:31:17 +11:00 |
|
Ryan Dick
|
e82393f7ed
|
Add FLUX Redux to starter models list.
|
2025-03-06 10:31:17 +11:00 |
|
Ryan Dick
|
d5211a8088
|
Add FluxRedux model type and probing logic.
|
2025-03-06 10:31:17 +11:00 |
|
Ryan Dick
|
3b095b5945
|
Add SigLIP starter model.
|
2025-03-06 10:31:17 +11:00 |
|
Ryan Dick
|
34959ef573
|
Add SigLIP model type and probing.
|
2025-03-06 10:31:17 +11:00 |
|
Billy
|
f2689598c0
|
Formatting
|
2025-03-06 09:11:00 +11:00 |
|
Ryan Dick
|
cc9d215a9b
|
Add endpoint for emptying the model cache. Also, adds a threading lock to the ModelCache to make it thread-safe.
|
2025-01-30 09:18:28 -05:00 |
|
Ryan Dick
|
f7315f0432
|
Make the default max RAM cache size more conservative.
|
2025-01-30 08:46:59 -05:00 |
|
Ryan Dick
|
229834a5e8
|
Performance optimizations for LoRAs applied on top of GGML-quantized tensors.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
6c919e1bca
|
Handle DoRA layer device casting when model is partially-loaded.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
5357d6e08e
|
Rename ConcatenatedLoRALayer to MergedLayerPatch. And other minor cleanup.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
7fef569e38
|
Update frontend graph building logic to support FLUX LoRAs that modify the T5 encoder weights.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
e7fb435cc5
|
Update DoRALayer with a custom get_parameters() override that 1) applies alpha scaling to delta_v, and 2) warns if the base model is incompatible.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
5d472ac1b8
|
Move quantized weight handling for patch layers up from ConcatenatedLoRALayer to CustomModuleMixin.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
28514ba59a
|
Update ConcatenatedLoRALayer to work with all sub-layer types.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
5ea7953537
|
Update GGMLTensor with ops necessary to work with ConcatenatedLoRALayer.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
0db6639b4b
|
Add FLUX OneTrainer model probing.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
b8eed2bdcb
|
Relax lora_layers_from_flux_diffusers_grouped_state_dict(...) so that it can work with more LoRA variants (e.g. hada)
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
1054283f5c
|
Fix bug in FLUX T5 Koyha-style LoRA key parsing.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
409b69ee5d
|
Fix typo in DoRALayer.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
206f261e45
|
Add utils for loading FLUX OneTrainer DoRA models.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
7eee4da896
|
Further updates to lora_model_from_flux_diffusers_state_dict() so that it can be re-used for OneTrainer LoRAs.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
908976ac08
|
Add support for LyCoris-style LoRA keys in lora_model_from_flux_diffusers_state_dict(). Previously, it only supported PEFT-style LoRA keys.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
dfa253e75b
|
Add utils for working with Kohya LoRA keys.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
4f369e3dfb
|
First draft of DoRALayer. Not tested yet.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
5bd6428fdd
|
Add is_state_dict_likely_in_flux_onetrainer_format() util function.
|
2025-01-28 14:51:35 +00:00 |
|
Ryan Dick
|
f88c1ba0c3
|
Fix bug with some LoRA variants when applied to a BnB NF4 quantized model. Note the previous commit which added a unit test to trigger this bug.
|
2025-01-22 09:20:40 +11:00 |
|
Ryan Dick
|
0cf51cefe8
|
Revise the logic for calculating the RAM model cache limit.
|
2025-01-16 23:46:07 +00:00 |
|
Ryan Dick
|
da589b3f1f
|
Memory optimization to load state dicts one module at a time in CachedModelWithPartialLoad when we are not storing a CPU copy of the state dict (i.e. when keep_ram_copy_of_weights=False).
|
2025-01-16 17:00:33 +00:00 |
|
Ryan Dick
|
36a3869af0
|
Add keep_ram_copy_of_weights config option.
|
2025-01-16 15:35:25 +00:00 |
|
Ryan Dick
|
c76d08d1fd
|
Add keep_ram_copy option to CachedModelOnlyFullLoad.
|
2025-01-16 15:08:23 +00:00 |
|
Ryan Dick
|
04087c38ce
|
Add keep_ram_copy option to CachedModelWithPartialLoad.
|
2025-01-16 14:51:44 +00:00 |
|
Ryan Dick
|
b2bb359d47
|
Update the model loading logic for several of the large FLUX-related models to ensure that the model is initialized on the meta device prior to loading the state dict into it. This helps to keep peak memory down.
|
2025-01-16 02:30:28 +00:00 |
|
Ryan Dick
|
b301785dc8
|
Normalize the T5 model identifiers so that a FLUX T5 or an SD3 T5 model can be used interchangeably.
|
2025-01-16 08:33:58 +11:00 |
|
Ryan Dick
|
607d19f4dd
|
We should not trust the value of since the model could be partially-loaded.
|
2025-01-07 19:22:31 -05:00 |
|