3 people dacaf5a400 Replace head_mapping params with num_kv_heads to attention kernel. (#1997) ...

Co-authored-by: wangguoya <wangguoya@baidu.com>
Co-authored-by: Yang Zhao <zhaoyangstar@foxmail.com>

2023-12-10 10:12:53 -08:00

..

quantization

Merge EmbeddedLLM/vllm-rocm into vLLM main (#1836)

2023-12-07 23:16:52 -08:00

__init__.py

Change the name to vLLM (#150)

2023-06-17 03:07:40 -07:00

activation.py

Add PyTorch-native implementation of custom layers (#1898)

2023-12-02 21:18:40 -08:00

attention.py

Replace head_mapping params with num_kv_heads to attention kernel. (#1997)

2023-12-10 10:12:53 -08:00

layernorm.py

Add PyTorch-native implementation of custom layers (#1898)

2023-12-02 21:18:40 -08:00

linear.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

rotary_embedding.py

Add PyTorch-native implementation of custom layers (#1898)

2023-12-02 21:18:40 -08:00

sampler.py

Refactor Worker & InputMetadata (#1843)

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

vocab_parallel_embedding.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