Benjamin Bossan 6030f9160e ENH Model and layer status for auxiliary modules (#2762) ...

Right now, get_model_status() and get_layer_status() only report on
BaseTunerLayers, but it would be helpful if they could also report
auxiliary modules. This PR now includes those.

To facilitate this, a few attributes and methods were added to
AuxiliaryTrainingWrapper and subclasses to make them more similar to
BaseTunerLayer (e.g. the adapter_layer_names attribute). These
attributes and methods were assumed to be present in the code that
determines the model and layer status.

2025-09-25 18:00:11 +02:00

..

bnb

align xpu behavior w/ cuda (#2551)

2025-06-02 17:23:42 +02:00

regression

TST XPU regression tests with deterministic (#2600)

2025-06-24 15:42:03 +02:00

__init__.py

TST: torch compile tests (#1725)

2024-05-17 18:03:27 +02:00

conftest.py

CI: Handle errors with MacOS and transformers (#2561)

2025-06-02 15:07:10 +02:00

test_adaption_prompt.py

ENH Llama-Adapters support for GPT2 (#2643)

2025-07-24 14:51:16 +02:00

test_arrow.py

Add Arrow + GenKnowSub to LoRA (#2644)

2025-09-08 14:21:37 +02:00

test_auto.py

FIX AutoPeftModels never reduce embedding size (#2427)

2025-03-14 14:17:31 +01:00

test_boft.py

ENH BOFT don't save boft_P buffer (#2050)

2024-09-13 13:56:47 +02:00

test_bufferdict.py

Trainable Tokens: Support for Weight Tying (#2399)

2025-03-06 14:09:01 +01:00

test_common_gpu.py

FEAT RoAd: 2D Rotary Adaptation (#2678)

2025-08-19 15:45:38 +02:00

test_config.py

TST Add missing configs to test_config.py (#2781)

2025-09-19 17:52:58 +02:00

test_cpt.py

FIX: CPT should not be tested with sequence classification (#2507)

2025-04-23 18:46:27 +02:00

test_custom_models.py

FIX PEFT layers expose in_features, out_features (#2784)

2025-09-25 17:59:45 +02:00

test_decoder_models.py

Support for Activated LoRA (#2609)

2025-09-03 18:26:50 +02:00

test_encoder_decoder_models.py

FEAT RoAd: 2D Rotary Adaptation (#2678)

2025-08-19 15:45:38 +02:00

test_feature_extraction_models.py

FEAT RoAd: 2D Rotary Adaptation (#2678)

2025-08-19 15:45:38 +02:00

test_gptqmodel.py

ENH Make OFT faster and more memory efficient (#2575)

2025-06-26 14:27:03 +02:00

test_gpu_examples.py

Add Arrow + GenKnowSub to LoRA (#2644)

2025-09-08 14:21:37 +02:00

test_helpers.py

MNT Use Python 3.9 as RUFF target version (#2483)

2025-04-08 16:44:55 +02:00

test_hub_features.py

Use hub_online_once in trainable token tests (#2701)

2025-08-05 12:58:55 +02:00

test_incremental_pca.py

TST: Add more HF Hub model caching (#2682)

2025-07-30 20:02:07 +02:00

test_initialization.py

FIX Small fixes to warning, like missing spaces (#2788)

2025-09-25 17:58:07 +02:00

test_integrations.py

ENH Add LoRA multihead attention module (#1324)

2025-01-08 17:35:43 +01:00

test_lora_megatron.py

TST Make tests pass on Cambricon MLUs (#1747)

2024-06-06 10:44:03 +02:00

test_lora_variants.py

Support for Activated LoRA (#2609)

2025-09-03 18:26:50 +02:00

test_lorafa.py

FEAT Add LoRA-FA to PEFT (#2468)

2025-04-10 10:53:19 +02:00

test_loraplus.py

TST Enable BNB tests on XPU (#2396)

2025-03-06 16:18:47 +01:00

test_low_level_api.py

Use hub_online_once in trainable token tests (#2701)

2025-08-05 12:58:55 +02:00

test_mapping.py

FIX #2295: Warn when user reloads modified model (#2306)

2025-01-07 18:10:07 +01:00

test_mixed.py

FIX: Multiple active adapters with auxiliary layers (#2758)

2025-08-29 17:54:19 +02:00

test_multitask_prompt_tuning.py

TST: Refactor remaining common tests to use pytest (#2491)

2025-05-02 11:19:32 +02:00

test_other.py

FIX Correctly determine no_split_modules (#2570)

2025-06-16 17:21:06 +02:00

test_poly.py

TST Use plain asserts in tests (#1448)

2024-02-14 16:43:47 +01:00

test_randlora.py

align xpu behavior w/ cuda (#2551)

2025-06-02 17:23:42 +02:00

test_seq_classifier.py

FEAT RoAd: 2D Rotary Adaptation (#2678)

2025-08-19 15:45:38 +02:00

test_shira.py

FEAT Add SHiRA Adapters (#2584)

2025-07-14 11:16:10 +02:00

test_stablediffusion.py

ENH Make OFT faster and more memory efficient (#2575)

2025-06-26 14:27:03 +02:00

test_target_parameters.py

FIX Multiple issues with target_parameters (#2710)

2025-08-12 13:59:29 +02:00

test_torch_compile.py

FEAT Add MiSS as a replacement for Bone. (#2604)

2025-08-01 18:37:20 +02:00

test_trainable_tokens.py

Use hub_online_once in trainable token tests (#2701)

2025-08-05 12:58:55 +02:00

test_tuners_utils.py

ENH Model and layer status for auxiliary modules (#2762)

2025-09-25 18:00:11 +02:00

test_vblora.py

align xpu behavior w/ cuda (#2551)

2025-06-02 17:23:42 +02:00

test_vera.py

align xpu behavior w/ cuda (#2551)

2025-06-02 17:23:42 +02:00

test_vision_models.py

ENH Make OFT faster and more memory efficient (#2575)

2025-06-26 14:27:03 +02:00

test_xlora.py

FIX X-LoRA forward hook issue during generate (#2761)

2025-09-08 13:46:31 +02:00

testing_common.py

ENH Model and layer status for auxiliary modules (#2762)

2025-09-25 18:00:11 +02:00

testing_utils.py

CHORE: Clean up config kwargs in custom model tests (#2736)

2025-08-19 11:55:25 +02:00