text-generation-inference/server/text_generation_server/utils
yuanwu 92a1e0fbae Aligin the source code with main branch 2.0.4
Signed-off-by: yuanwu <yuan.wu@intel.com>
2024-09-24 03:06:55 +00:00
..
__init__.py Aligin the source code with main branch 2.0.4 2024-09-24 03:06:55 +00:00
convert.py Force weights_only (before fully breaking pickle files anyway). (#1710) 2024-04-25 15:10:53 +03:00
dist.py Aligin the source code with main branch 2.0.4 2024-09-24 03:06:55 +00:00
flash_attn_triton.py MI300 compatibility (#1764) 2024-07-17 05:36:58 +00:00
flash_attn.py reenable xpu for tgi (#1939) 2024-07-17 05:36:58 +00:00
hub.py Fixing the download strategy for ibm-fms (#1917) 2024-07-17 05:36:58 +00:00
import_utils.py reenable xpu for tgi (#1939) 2024-07-17 05:36:58 +00:00
log.py v1.3.4 2024-04-22 09:08:34 +03:00
logits_process.py Aligin the source code with main branch 2.0.4 2024-09-24 03:06:55 +00:00
paged_attention.py MI300 compatibility (#1764) 2024-07-17 05:36:58 +00:00
peft.py fix: fix local loading for .bin models (#1419) 2024-04-22 09:17:52 +03:00
speculate.py chore: formatting 2024-04-18 16:26:00 +03:00
tokens.py Aligin the source code with main branch 2.0.4 2024-09-24 03:06:55 +00:00
watermark.py Aligin the source code with main branch 2.0.4 2024-09-24 03:06:55 +00:00
weights.py Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00