ColossalAI/colossalai/inference/modeling
Yuanheng Zhao 8754abae24 [Fix] Fix & Update Inference Tests (compatibility w/ main) 2024-05-05 16:28:56 +00:00
..
layers [Inference] Adapt Baichuan2-13B TP (#5659) 2024-04-30 15:47:07 +08:00
models [Fix] Fix & Update Inference Tests (compatibility w/ main) 2024-05-05 16:28:56 +00:00
policy [Inference] Adapt Baichuan2-13B TP (#5659) 2024-04-30 15:47:07 +08:00
__init__.py [doc] updated inference readme (#5343) 2024-02-02 14:31:10 +08:00