Noam Gat 555bdcc5a3 Added logits processor API to sampling params (#1469)

2023-11-03 14:12:15 -07:00

..

layers

Added logits processor API to sampling params (#1469)

2023-11-03 14:12:15 -07:00

models

Remove MPTConfig (#1529)

2023-11-01 15:29:05 -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

Remove MPTConfig (#1529)

2023-11-01 15:29:05 -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