Commit Graph

13485 Commits

Author SHA1 Message Date
Ryan Dick
e0c2b13558 Remove unused layer_key property from LoRALayerBase. 2024-09-10 15:17:12 +00:00
Ryan Dick
475e92e305 Consolidate all LoRA patching logic in the LoraPatcher. 2024-09-10 15:09:46 +00:00
Ryan Dick
99a12fb604 Rename peft -> lora in a bunch of places. 2024-09-10 14:00:41 +00:00
Ryan Dick
1a12c48f6e Rename lora.py -> lora_model_raw.py. 2024-09-10 13:54:17 +00:00
Ryan Dick
f1c8fd16b5 Rename peft/ -> lora/ 2024-09-10 13:52:06 +00:00
Ryan Dick
f44a6a7fb2 Genera cleanup/documentation. 2024-09-09 21:52:05 +00:00
Ryan Dick
da780c2243 Add a check that all keys are handled in the FLUX Diffusers LoRA loading code. 2024-09-09 21:13:04 +00:00
Ryan Dick
742f6781d5 Add model probe support for FLUX LoRA models in Diffusers format. 2024-09-09 21:00:40 +00:00
Ryan Dick
d2ffabf276 Add utility test function for creating a dummy state_dict. 2024-09-09 21:00:40 +00:00
Ryan Dick
39e28d5e24 Add is_state_dict_likely_in_flux_diffusers_format(...) function with unit test. 2024-09-09 21:00:40 +00:00
Ryan Dick
3f7c233f4d Add unit test for lora_model_from_flux_diffusers_state_dict(...). 2024-09-09 21:00:40 +00:00
Ryan Dick
af75a8ea99 First draft of lora_model_from_flux_diffusers_state_dict(...). 2024-09-09 21:00:40 +00:00
Ryan Dick
ddfe57e648 (minor) Rename test file. 2024-09-09 21:00:40 +00:00
Ryan Dick
7e8dd9e8ed Add ConcatenateLoRALayer class. 2024-09-09 21:00:40 +00:00
Ryan Dick
50f8d6db1b WIP on supporting diffusers format FLUX LoRAs. 2024-09-09 21:00:40 +00:00
Ryan Dick
961fbe1ba4 Rename flux_kohya_lora_conversion_utils.py 2024-09-09 21:00:40 +00:00
Ryan Dick
7342d18734 Fixup FLUX LoRA unit tests. 2024-09-09 21:00:40 +00:00
Ryan Dick
554a4dc592 WIP 2024-09-09 21:00:40 +00:00
Ryan Dick
90e486c976 WIP - add invocations to support FLUX LORAs. 2024-09-09 21:00:35 +00:00
Ryan Dick
ceb5d50568 Get probing of FLUX LoRA kohya models working. 2024-09-09 20:58:27 +00:00
Ryan Dick
e4cca62a90 Add utility function for detecting whether a state_dict is in the FLUX kohya LoRA format. 2024-09-09 20:58:27 +00:00
Ryan Dick
008f672e47 Update convert_flux_kohya_state_dict_to_invoke_format() to raise an exception if an unexpected key is encountered, and add a corresponding unit test. 2024-09-09 20:58:27 +00:00
Ryan Dick
53ae86068c Move the responsibilities of 1) state_dict loading from file, and 2) SDXL lora key conversions, out of LoRAModelRaw and into LoRALoader. 2024-09-09 20:58:27 +00:00
Ryan Dick
7de3c1943f Remove unused LoRAModelRaw.name attribute. 2024-09-09 20:58:27 +00:00
Ryan Dick
a0f36dea31 Fix type errors in sdxl_conversion_utils.py 2024-09-09 20:58:27 +00:00
Ryan Dick
69a2f8d53d Start moving SDXL-specific LoRA conversions out of the general-purpose LoRAModelRaw class. 2024-09-09 20:58:27 +00:00
Ryan Dick
3f2a61e0a6 Get convert_flux_kohya_state_dict_to_invoke_format(...) working, with unit tests. 2024-09-09 20:58:27 +00:00
Ryan Dick
f45b925bbf WIP - FLUX LoRA conversion logic. 2024-09-09 20:58:27 +00:00
Ryan Dick
d2870a512d Add state_dict keys for two FLUX LoRA formats to be used in unit tests. 2024-09-09 20:58:27 +00:00
Ryan Dick
e1e5f970e6 Move lora.py to peft/ subdir. 2024-09-09 20:58:27 +00:00
Ryan Dick
0712684dc9 Split PEFT layer implementations into separate files. 2024-09-09 20:58:27 +00:00
Brandon Rising
edae8a1617 Update to reflect an alpha release v5.0.0.a1 2024-09-09 13:50:15 -04:00
Brandon Rising
9c1cf3e860 chore: 5.0.0.dev14 version bump 2024-09-09 13:50:15 -04:00
psychedelicious
b6cef9d440 fix(ui): do not clear buffer on escape if filtering/transforming 2024-09-09 23:40:38 +10:00
psychedelicious
ebb92bee26 fix(ui): use reactive entity adapter hooks, fix one-behind issue 2024-09-09 23:40:38 +10:00
psychedelicious
d6c553ca5e chore(ui): lint 2024-09-09 23:17:41 +10:00
psychedelicious
8b6512cc90 fix(ui): stale rect used in getVisibleRect (partial fix)
Need to figure out why the rect isn't reset when the entity is reset. Probably just needs some special handling.
2024-09-09 23:17:41 +10:00
psychedelicious
a6b998c125 feat(ui): move fit bbox to layers button to toolbar 2024-09-09 23:17:41 +10:00
psychedelicious
5275782533 feat(ui): move add layer menu to selected entity action bar 2024-09-09 23:17:41 +10:00
psychedelicious
ede3bd8e64 feat(ui): default canvas state includes bookmarked inpaint mask 2024-09-09 23:17:41 +10:00
psychedelicious
da2583b894 feat(ui): shift+c clears regional guidance 2024-09-09 23:17:41 +10:00
psychedelicious
9210970130 fix(ui): preview not updating after reset 2024-09-09 23:17:41 +10:00
psychedelicious
2a022a811c feat(ui): selected entity alert 2024-09-09 23:17:41 +10:00
psychedelicious
1a53e8dc5c feat(ui): swap gallery and layer tabs 2024-09-09 23:17:41 +10:00
psychedelicious
4e12e23b69 feat(ui): tweak left panel size 2024-09-09 23:17:41 +10:00
psychedelicious
fd56b35982 fix(ui): vae layout 2024-09-09 23:17:41 +10:00
psychedelicious
71e0abe653 fix(ui): preview image squished when editing layer title 2024-09-09 23:17:41 +10:00
psychedelicious
56956ccf78 tidy(ui): remove extraneous fallback in QueueCountBadge 2024-09-09 23:17:41 +10:00
psychedelicious
6d46d82028 feat(ui): do not render anything except current content
This makes it a bit slower to switch tabs but also eliminates a whole class of bugs related to rendered but hidden components.
2024-09-09 23:17:41 +10:00
psychedelicious
3ed29a16a8 feat(ui): reworked layout (wip) 2024-09-09 23:17:41 +10:00