Roy f1c0fc3919 Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

..

layers

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

models

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

parallel_utils

Fix dist.broadcast stall without group argument (#3408)

2024-03-14 23:25:05 -07:00

__init__.py

[Neuron] Support inference with transformers-neuronx (#2569)

2024-02-28 09:34:34 -08:00

guided_decoding.py

Support arbitrary json_object in OpenAI and Context Free Grammar (#3211)

2024-03-16 13:35:27 -07:00

guided_logits_processors.py

Support arbitrary json_object in OpenAI and Context Free Grammar (#3211)

2024-03-16 13:35:27 -07:00

input_metadata.py

[1/n][Chunked Prefill] Refactor input query shapes (#3236)

2024-03-20 14:46:05 -07:00

model_loader.py

[Neuron] Support inference with transformers-neuronx (#2569)

2024-02-28 09:34:34 -08:00

neuron_model_loader.py

Re-enable the 80 char line width limit (#3305)

2024-03-10 19:49:14 -07:00

sampling_metadata.py

[1/n] Triton sampling kernel (#3186)

2024-03-20 14:45:08 -07:00

utils.py

[Neuron] Support inference with transformers-neuronx (#2569)

2024-02-28 09:34:34 -08:00

weight_utils.py

Move model filelocks from /tmp/ to ~/.cache/vllm/locks/ dir (#3241)

2024-03-08 13:33:10 -08:00