Jinzhen Lin 99caa49106 [Kernel] add bfloat16 support for gptq marlin kernel (#4788)

2024-05-16 09:55:29 -04:00

..

fused_moe

[Model] Snowflake arctic model implementation (#4652)

2024-05-09 22:37:14 +00:00

ops

[Mypy] Part 3 fix typing for nested directories for most of directory (#4161)

2024-04-22 21:32:44 -07:00

quantization

[Kernel] add bfloat16 support for gptq marlin kernel (#4788)

2024-05-16 09:55:29 -04:00

__init__.py

Change the name to vLLM (#150)

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

activation.py

[Misc]Add customized information for models (#4132)

2024-04-30 21:18:14 -07:00

layernorm.py

[Misc]Add customized information for models (#4132)

2024-04-30 21:18:14 -07:00

linear.py

[Misc]Add customized information for models (#4132)

2024-04-30 21:18:14 -07:00

logits_processor.py

[Misc]Add customized information for models (#4132)

2024-04-30 21:18:14 -07:00

pooler.py

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

rejection_sampler.py

[Core][2/N] Model runner refactoring part 2. Combine prepare prefill / decode to a single API (#4681)

2024-05-15 14:00:10 +09:00

rotary_embedding.py

[CI] Make mistral tests pass (#4596)

2024-05-08 08:44:35 -07:00

sampler.py

[CORE] Improvement in ranks code (#4718)

2024-05-12 17:47:47 -07:00

vocab_parallel_embedding.py

[Misc]Add customized information for models (#4132)

2024-04-30 21:18:14 -07:00