Philipp MoritzandRoy 317b29de0f Remove Yi model definition, please use LlamaForCausalLM instead (#2854) ...

Co-authored-by: Roy <jasonailu87@gmail.com>

2024-02-13 14:22:22 -08:00

..

layers

Revert "Refactor llama family models (#2637)" (#2851)

2024-02-13 09:24:59 -08:00

models

Remove Yi model definition, please use LlamaForCausalLM instead (#2854)

2024-02-13 14:22:22 -08:00

parallel_utils

Use CuPy for CUDA graphs (#2811)

2024-02-13 11:32:06 -08:00

__init__.py

Refactor Worker & InputMetadata (#1843)

2023-11-29 22:16:37 -08:00

input_metadata.py

Support FP8-E5M2 KV Cache (#2279)

2024-01-28 16:43:54 -08:00

model_loader.py

Remove hardcoded device="cuda" to support more devices (#2503)

2024-02-01 15:46:39 -08:00

sampling_metadata.py

Use NCCL instead of ray for control-plane communication to remove serialization overhead (#2221)

2024-01-03 11:30:22 -08:00

utils.py

TP/quantization/weight loading refactor part 2 - Refactor quantized linear logic and extend quantization support to all models (#1622)

2023-11-15 22:50:41 -08:00

weight_utils.py

Use revision when downloading the quantization config file (#2697)

2024-02-01 15:41:58 -08:00