Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
hf_text-generation-inference/server/text_generation_server/layers
History
Dhruv Srikanth 6c715f8183
[Bug Fix] Update torch import reference in bnb quantization (#1902) 
# What does this PR do?

Fixes `Import Error` occurring from mismatch of usage between
torch.nn.Module and nn.Module.
 		2024-05-15 21:08:32 +02:00
..
awq Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
gptq Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
__init__.py MLPSpeculator. (#1865) 2024-05-14 12:33:18 +02:00
bnb.py [Bug Fix] Update torch import reference in bnb quantization (#1902) 2024-05-15 21:08:32 +02:00
conv.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
eetq.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
fp8.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
layernorm.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
linear.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
medusa.py MLPSpeculator. (#1865) 2024-05-14 12:33:18 +02:00
mlp.py MLPSpeculator. (#1865) 2024-05-14 12:33:18 +02:00
rotary.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
speculative.py MLPSpeculator. (#1865) 2024-05-14 12:33:18 +02:00
tensor_parallel.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00