ColossalAI/applications
YeAnbang 26d859f68e
[feat] Support DAPO (#6263)
* update help information

* update style

* fix

* minor fix

* support PP training

* add pp support

* remove unused code

* address conversation

* fix memory leakage support tp+pp

* move empty cache

* move empty cache

* add DAPO support

* remove format reward

* fix filtering, still buggy

* small fix

* add DAPO support

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* tested multi-node training; fix bind_batch bug

* fix conversation; support sleep mode

* support reusing excessive samples

* add dynamic batching control flag

* add dynamic batching control flag

* refactored

* fix logging

---------

Co-authored-by: Tong Li <tong.li35271158@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2025-04-25 17:39:17 +08:00
..
Colossal-LLaMA [Device]Support npu (#6159) 2024-12-17 15:42:39 +08:00
ColossalChat [feat] Support DAPO (#6263) 2025-04-25 17:39:17 +08:00
ColossalEval [ColossalEval] support for vllm (#6056) 2024-09-18 17:09:45 +08:00
ColossalMoE [MoE/ZeRO] Moe refactor with zero refactor (#5821) 2024-06-28 14:00:08 +08:00
ColossalQA [pre-commit.ci] pre-commit autoupdate (#5572) 2024-07-01 17:16:41 +08:00
README.md [Hotfix] README link (#5966) 2024-08-08 18:04:47 +08:00

Applications

This directory contains the applications that are powered by Colossal-AI.

The list of applications include:

  • Open-Sora: Revealing Complete Model Parameters, Training Details, and Everything for Sora-like Video Generation Models
  • ColossalChat: Replication of ChatGPT with RLHF.
  • Colossal-LLaMA: Continual Pre-training and Supervisied Fine-tuning of LLaMA2 / LLaMA3.
  • ColossalEval: Evaluation Pipeline for LLMs.
  • FastFold: Optimizing AlphaFold (Biomedicine) Training and Inference on GPU Clusters.
  • ColossalQA: Document Retrieval Conversation System
  • SwiftInfer: Breaks the Length Limit of LLM Inference for Multi-Round Conversations

Please note that the Chatbot application is migrated from the original ChatGPT folder.

You can find more example code for base models and functions in the Examples directory.