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