Mor ZusmanandMor Zusman 07ab160741 [Model][Jamba] Mamba cache single buffer (#6739) ...

Co-authored-by: Mor Zusman <morz@ai21.com>

2024-08-09 10:07:06 -04:00

..

__init__.py

Change the name to vLLM (#150)

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

cache_engine.py

[Model] Jamba support (#4115)

2024-07-02 23:11:29 +00:00

cpu_model_runner.py

[Bugfix] Fix broadcasting logic for multi_modal_kwargs (#6836)

2024-07-31 10:38:45 +08:00

cpu_worker.py

[Hardware] [Intel] Enable Multiprocessing and tensor parallel in CPU backend and update documentation (#6125)

2024-07-26 13:50:10 -07:00

embedding_model_runner.py

[Bugfix] Fix broadcasting logic for multi_modal_kwargs (#6836)

2024-07-31 10:38:45 +08:00

enc_dec_model_runner.py

[Core] Subclass ModelRunner to support cross-attention & encoder sequences (towards eventual encoder/decoder model support) (#4942)

2024-08-06 16:51:47 -04:00

model_runner_base.py

[BugFix] Fix use of per-request seed with pipeline parallel (#6698)

2024-07-30 10:40:08 -07:00

model_runner.py

[Model][Jamba] Mamba cache single buffer (#6739)

2024-08-09 10:07:06 -04:00

neuron_model_runner.py

[Bugfix] Fix broadcasting logic for multi_modal_kwargs (#6836)

2024-07-31 10:38:45 +08:00

neuron_worker.py

[Core] Pipeline Parallel Support (#4412)

2024-07-02 10:58:08 -07:00

openvino_model_runner.py

[Bugfix] Fix broadcasting logic for multi_modal_kwargs (#6836)

2024-07-31 10:38:45 +08:00

openvino_worker.py

[core][distributed] support n layers % pp size != 0 (#6115)

2024-07-03 16:40:31 -07:00

tpu_model_runner.py

[TPU] Fix greedy decoding (#6933)

2024-07-30 02:06:29 -07:00

tpu_worker.py

[Bugfix][TPU] Set readonly=True for non-root devices (#6980)

2024-07-31 00:19:28 -07:00

utils.py

[Core] Subclass ModelRunner to support cross-attention & encoder sequences (towards eventual encoder/decoder model support) (#4942)

2024-08-06 16:51:47 -04:00

worker_base.py

[Core] Pipeline parallel with Ray ADAG (#6837)

2024-08-02 13:55:40 -07:00

worker.py

[Core] Support serving encoder/decoder models (#7258)

2024-08-09 10:39:41 +08:00

xpu_model_runner.py

[Bugfix] Fix broadcasting logic for multi_modal_kwargs (#6836)

2024-07-31 10:38:45 +08:00

xpu_worker.py

[ci] set timeout for test_oot_registration.py (#7082)

2024-08-02 10:03:24 -07:00