huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-24 00:12:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
692f8ddb69
text-generation-inference/server/text_generation_server/layers
History
drbh bdc47394d2 feat: support phi3.5 moe (#2479) 
* feat: support phi3.5 moe model loading

* fix: prefer llama base model and improve rotary logic

* feat: return reasonable generation and add integration test

* fix: run lint and update docs

* fix: rerun lint for openapi docs

* fix: prefer do_sample false unless temp is set by user, and update chat tests

* fix: small typo adjustments

* fix: consolidate long rope paths

* fix: revert greedy by default and test changes

* Vendor configuration so that we don't have to `trust_remote_code`

* Use SparseMoELayer

* Add support for dense MoE

* Some type annotations

* Add the usual model tests

* Ruff.

---------

Co-authored-by: Daniël de Kok <me@danieldk.eu>
Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>
2024-10-25 09:12:03 +00:00
..
attention Update ROCM libs and improvements (#2579) 2024-10-25 09:01:04 +00:00
awq feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
gptq Add some missing modification of 2.3.0 because of conflict 2024-09-25 07:49:49 +00:00
marlin Handle GPTQ-Marlin loading in GPTQMarlinWeightLoader (#2300) 2024-09-25 05:55:39 +00:00
moe Add support for GPTQ-quantized MoE models using MoE Marlin (#2557) 2024-10-25 09:07:52 +00:00
__init__.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
bnb.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
conv.py Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00
eetq.py feat(fp8): use fbgemm kernels and load fp8 weights directly (#2248) 2024-09-25 05:30:41 +00:00
exl2.py Add support for Deepseek V2 (#2224) 2024-09-25 05:27:40 +00:00
fp8.py Add support for scalar FP8 weight scales (#2550) 2024-10-25 09:01:04 +00:00
layernorm.py Removing IPEX_AVAIL. (#2115) 2024-09-24 03:52:23 +00:00
linear.py Update ROCM libs and improvements (#2579) 2024-10-25 09:01:04 +00:00
lora.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
medusa.py Prefix caching (#2402) 2024-09-25 06:10:59 +00:00
mlp.py Tied embeddings in MLP speculator. (#2473) 2024-09-25 06:13:11 +00:00
rotary.py feat: support phi3.5 moe (#2479) 2024-10-25 09:12:03 +00:00
speculative.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
tensor_parallel.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00