text-generation-inference/server/text_generation_server
Dhruv Srikanth b1d370e062 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-07-17 05:36:58 +00:00
..
layers Update torch import reference in bnb quantization (#1902) 2024-07-17 05:36:58 +00:00
models Removing accepted ids in the regular info logs, downgrade to debug. (#1898) 2024-07-17 05:36:58 +00:00
pb chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
utils Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00
__init__.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
cache.py fix(server): decrease memory fragmentation (#557) 2023-07-06 14:28:33 +02:00
cli.py Fp8 Support (#1726) 2024-04-25 17:58:11 +03:00
habana_quantization_env.py Add Habana copyright header (#122) 2024-04-08 18:06:21 +02:00
interceptor.py v2.0.0 (#1736) 2024-04-26 07:42:52 +00:00
server.py Better graceful shutdown. (#1827) 2024-06-10 14:00:26 +03:00
tgi_service.py Speculative (#1308) 2024-04-18 12:39:39 +00:00
tracing.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00