Ryan Dick 2335b70dba Add is_state_dict_likely_in_flux_onetrainer_format() util function.

2025-01-23 23:11:55 +00:00

..

flux

Infer the clip_extra_context_tokens param from the state dict for FLUX XLabs IP-Adapter V2 models.

2024-11-18 17:06:53 -08:00

ip_adapter

Create a UNetAttentionPatcher for patching UNet models with CustomAttnProcessor2_0 modules.

2024-04-09 08:12:12 -04:00

model_manager

Add unit tests for LoKR patch layers. The new tests trigger a bug when LoKR layers are applied to BnB-quantized layers (also impacts several other LoRA variant types).

2025-01-22 09:20:40 +11:00

patches

Add is_state_dict_likely_in_flux_onetrainer_format() util function.

2025-01-23 23:11:55 +00:00

quantization

Reduce peak memory used for unit tests.

2024-12-24 14:32:11 +00:00

stable_diffusion

Add unit tests for ExtensionsManager and ExtensionBase.

2024-07-19 14:15:46 -04:00

tiles

In CalculateImageTilesEvenSplitInvocation to have overlap_fraction becomes just overlap. This is now in pixels rather than as a fraction of the tile size.

2023-12-17 15:10:50 +00:00

util

Add build_line(...) util function.

2024-09-20 01:01:37 +00:00

__init__.py

POC of a test that depends on models.

2023-10-05 15:35:58 -04:00