Chansung Park 27f7dbf006 Fix BitsAndBytes JSON Serializable (#191) ...

* Update run_sft.py

* fix BitsAndBytes JSON serializable

* get_quantization_config to return dict

* to_dict() for load_in_8bit too

* convert quant test to use dict subscriptions instead of dot syntax

* Remove torch

---------

Co-authored-by: lewtun <lewis.c.tunstall@gmail.com>

2024-08-20 15:02:18 +02:00

..

fixtures

update API to use latest TRL (#182)

2024-07-30 09:16:25 +02:00

__init__.py

Add tests folder

2023-10-26 09:54:20 +00:00

test_configs.py

Add config tests

2023-11-10 10:00:05 +00:00

test_data.py

Use ungated models for unit tests (#196)

2024-08-19 12:33:11 +02:00

test_decontaminate.py

🌟 (#135)

2024-03-12 17:22:21 +01:00

test_model_utils.py

Fix BitsAndBytes JSON Serializable (#191)

2024-08-20 15:02:18 +02:00