Jee Li 67005a07bc

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

Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>

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

..

__init__.py

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

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

fully_sharded_layers.py

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

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

layers.py

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

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

lora.py

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

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

models.py

[Model] LoRA gptbigcode implementation (#3949)

2024-05-22 13:58:59 -07:00

punica.py

[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)

2024-06-09 16:23:30 -04:00

request.py

[Lora] Support long context lora (#4787)

2024-05-18 16:05:23 +09:00

utils.py

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

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

worker_manager.py

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

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