Cade Daniel 18bfcdd05c [Speculative decoding 2/9] Multi-step worker for draft model (#2424)

2024-01-21 16:31:47 -08:00

..

layers

[Experimental] Prefix Caching Support (#1669)

2024-01-17 16:32:10 -08:00

models

fix stablelm.py tensor-parallel-size bug (#2482)

2024-01-18 09:39:46 -08:00

parallel_utils

[Speculative decoding 2/9] Multi-step worker for draft model (#2424)

2024-01-21 16:31:47 -08:00

__init__.py

Refactor Worker & InputMetadata (#1843)

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

input_metadata.py

[Experimental] Prefix Caching Support (#1669)

2024-01-17 16:32:10 -08:00

model_loader.py

Implement lazy model loader (#2044)

2023-12-12 22:21:45 -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

[Bugfix] fix load local safetensors model (#2512)

2024-01-19 16:26:16 -08:00