divyanshuaggarwal 170cdb5be9 Add Post_model_load, post_lora_load, post_train, post_train_unload function calls (#2539) ...

* Update train.py

add post_model_load and post_lora_load model calss.

* Update train.py

add post_train and post_train_unload function calls

* Update train.py

* Update base.py

* Update train.py

* chore: lint

* clarify plugin hooks

* Update src/axolotl/integrations/base.py

Co-authored-by: Dan Saunders <danjsaund@gmail.com>

* Update src/axolotl/utils/models.py

Co-authored-by: Dan Saunders <danjsaund@gmail.com>

* Update src/axolotl/utils/models.py

Co-authored-by: Dan Saunders <danjsaund@gmail.com>

* Update src/axolotl/integrations/base.py

Co-authored-by: Dan Saunders <danjsaund@gmail.com>

* Update models.py

* Update models.py

* remove extra call to post_model_load

* chore: lint

* add test for hooks and gc trainer

* disable duplicated code check for test

* fix the path and add better handling

---------

Co-authored-by: Wing Lian <wing@axolotl.ai>
Co-authored-by: Dan Saunders <danjsaund@gmail.com>

2025-04-28 10:10:28 -04:00

..

__init__.py

add e2e smoke tests for llama liger integration (#1884)

2024-09-01 19:29:37 -04:00

test_cut_cross_entropy.py

make sure to validate the config before normalizing so defaults get set (#2554)

2025-04-24 13:01:43 -04:00

test_hooks.py

Add Post_model_load, post_lora_load, post_train, post_train_unload function calls (#2539)

2025-04-28 10:10:28 -04:00

test_kd.py

fix: minor patches for multimodal (#2441)

2025-03-31 13:40:12 +07:00

test_liger.py

make sure to validate the config before normalizing so defaults get set (#2554)

2025-04-24 13:01:43 -04:00