Joe Runde ba4994443a [Kernel] Add punica dimensions for Granite 3b and 8b (#5930) ...

Signed-off-by: Joe Runde <joe@joerun.de>

2024-06-29 10:48:25 +08:00

..

data

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

__init__.py

[Experimental] Add multi-LoRA support (#1804)

2024-01-23 15:26:37 -08:00

conftest.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_baichuan.py

[Bugfix] Add fully sharded layer for QKVParallelLinearWithLora (#5665)

2024-06-21 04:46:28 +00:00

test_chatglm3.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_gemma.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_layer_variation.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_layers.py

[Bugfix] Add fully sharded layer for QKVParallelLinearWithLora (#5665)

2024-06-21 04:46:28 +00:00

test_llama.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_long_context.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_lora_checkpoints.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_lora_manager.py

[LoRA] Add support for pinning lora adapters in the LRU cache (#5603)

2024-06-21 15:42:46 -07:00

test_lora.py

[Experimental] Add multi-LoRA support (#1804)

2024-01-23 15:26:37 -08:00

test_mixtral.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_phi.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_punica.py

[Kernel] Add punica dimensions for Granite 3b and 8b (#5930)

2024-06-29 10:48:25 +08:00

test_quant_model.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_tokenizer_group.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

test_utils.py

[Misc] Improve error message when LoRA parsing fails (#5194)

2024-06-10 19:38:49 +08:00

test_worker.py

[Core] Refactor model loading code (#4097)

2024-04-16 11:34:39 -07:00

utils.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00