Antoni Baum 15f5632365 Delay GPU->CPU sync in sampling (#1337)

2023-10-30 09:01:34 -07:00

..

layers

Delay GPU->CPU sync in sampling (#1337)

2023-10-30 09:01:34 -07:00

models

Fix bias in InternLM (#1501)

2023-10-29 16:24:18 -07:00

parallel_utils

Implement prompt logprobs & Batched topk for computing logprobs (#1328)

2023-10-16 10:56:50 -07:00

quantization_utils

Support SqueezeLLM (#1326)

2023-10-21 23:14:59 -07:00

__init__.py

[Quality] Add code formatter and linter (#326)

2023-07-03 11:31:55 -07:00

input_metadata.py

Delay GPU->CPU sync in sampling (#1337)

2023-10-30 09:01:34 -07:00

model_loader.py

Fix the issue for AquilaChat2-* models (#1339)

2023-10-13 11:51:29 -07:00

utils.py

TP/quantization/weight loading refactor part 1 - Simplify parallel linear logic (#1181)

2023-10-02 15:36:09 -07:00

weight_utils.py

Add blacklist in model checkpoint (#1325)

2023-10-12 01:05:37 -07:00