Woosuk Kwon 929b4f2973 Add LoRA support for Gemma (#3050)

2024-02-28 13:03:28 -08:00

..

layers

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

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

models

Add LoRA support for Gemma (#3050)

2024-02-28 13:03:28 -08:00

parallel_utils

chore(vllm): codespell for spell checking (#2820)

2024-02-21 18:56:01 -08:00

__init__.py

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

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

input_metadata.py

Support FP8-E5M2 KV Cache (#2279)

2024-01-28 16:43:54 -08:00

model_loader.py

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

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

neuron_model_loader.py

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

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

sampling_metadata.py

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

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

utils.py

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

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

weight_utils.py

Use revision when downloading the quantization config file (#2697)

2024-02-01 15:41:58 -08:00