huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-23 07:52:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
57ba035a61
text-generation-inference/server/text_generation_server/layers
History
drbh 57ba035a61 fix: use path inside of speculator config (#1935) 
This PR access the path on the speculator similar to
`MLPSpeculatorHead.load` and `MedusaHeadV1.load`

these changes resolves this error locally when loading a `MedusaHeadV2`
```
TypeError: expected str, bytes or os.PathLike object, not dict
```
2024-07-17 05:36:58 +00:00
..
awq Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00
gptq Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00
__init__.py MLPSpeculator. (#1865) 2024-07-17 05:36:58 +00:00
bnb.py Update torch import reference in bnb quantization (#1902) 2024-07-17 05:36:58 +00:00
conv.py Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00
eetq.py Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00
fp8.py Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00
layernorm.py MI300 compatibility (#1764) 2024-07-17 05:36:58 +00:00
linear.py MI300 compatibility (#1764) 2024-07-17 05:36:58 +00:00
medusa.py fix: use path inside of speculator config (#1935) 2024-07-17 05:36:58 +00:00
mlp.py MLPSpeculator. (#1865) 2024-07-17 05:36:58 +00:00
rotary.py MI300 compatibility (#1764) 2024-07-17 05:36:58 +00:00
speculative.py MLPSpeculator. (#1865) 2024-07-17 05:36:58 +00:00
tensor_parallel.py Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00