Shu Wang 9e96f56efb

Allocate kv_cache with stride order (#16605) ...

Signed-off-by: shuw <shuw@nvidia.com>

2025-04-25 22:03:31 -07:00

..

__init__.py

Change the name to vLLM (#150)

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

cache_engine.py

Allocate kv_cache with stride order (#16605)

2025-04-25 22:03:31 -07:00

cpu_enc_dec_model_runner.py

[Chore] Remove Sampler from Model Code (#17084)

2025-04-24 02:49:33 -07:00

cpu_model_runner.py

Use Transformers helper get_text_config() instead of checking for text_config (#17105)

2025-04-25 08:47:35 -07:00

cpu_pooling_model_runner.py

Remove unused kwargs from model definitions (#13555)

2025-02-24 17:13:52 -08:00

cpu_worker.py

[CPU][Bugfix] Using custom allreduce for CPU backend (#15934)

2025-04-02 07:46:47 -07:00

enc_dec_model_runner.py

[Chore] Remove Sampler from Model Code (#17084)

2025-04-24 02:49:33 -07:00

hpu_model_runner.py

Use Transformers helper get_text_config() instead of checking for text_config (#17105)

2025-04-25 08:47:35 -07:00

hpu_worker.py

[PP] Correct cache size check (#13873)

2025-02-27 17:47:29 +08:00

model_runner_base.py

[Bugfix] Do not crash V0 engine on input errors (#13101)

2025-02-26 19:07:29 +08:00

model_runner.py

Use Transformers helper get_text_config() instead of checking for text_config (#17105)

2025-04-25 08:47:35 -07:00

multi_step_hpu_worker.py

[Hardware][Intel-Gaudi] Multi-step scheduling implementation for HPU (#12779)

2025-04-11 07:38:36 -07:00

multi_step_model_runner.py

[Chore] Remove Sampler from Model Code (#17084)

2025-04-24 02:49:33 -07:00

multi_step_tpu_worker.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

multi_step_worker.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

neuron_model_runner.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

neuron_worker.py

[Misc] Improve LoRA spelling (#13831)

2025-02-25 23:43:01 -08:00

pooling_model_runner.py

Remove unused kwargs from model definitions (#13555)

2025-02-24 17:13:52 -08:00

tpu_model_runner.py

Remove unused kwargs from model definitions (#13555)

2025-02-24 17:13:52 -08:00

tpu_worker.py

[TPU] support disabling xla compilation cache (#15567)

2025-03-27 00:09:28 +00:00

utils.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

worker_base.py

[RLHF] use worker_extension_cls for compatibility with V0 and V1 (#14185)

2025-03-07 00:32:46 +08:00

worker.py

Restore buffers when wake up from level 2 sleep (#16564) (#16889)

2025-04-21 20:18:28 +08:00

xpu_model_runner.py

[Chore] Remove Sampler from Model Code (#17084)

2025-04-24 02:49:33 -07:00

xpu_worker.py

[Misc] Improve LoRA spelling (#13831)

2025-02-25 23:43:01 -08:00