github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-19 09:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

update readme (#6116)

Browse Source
This commit is contained in:
Tong Li 2024-11-06 17:24:08 +08:00 committed by GitHub
parent a15ab139ad
commit 7a60161035
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
applications/ColossalChat/README.md
View File

@ -284,7 +284,7 @@ For more details, see [`inference/`](https://github.com/hpcaitech/ColossalAI/tre
## O1 Journey ## O1 Journey
### Inference with Self-refined MCTS ### Inference with Self-refined MCTS
We provide the implementation of MCT Self-Refine (MCTSr) algorithm, an innovative integration of Large Language Models with Monte Carlo Tree Search. We provide the implementation of MCT Self-Refine (MCTSr) algorithm, an innovative integration of Large Language Models with Monte Carlo Tree Search.
To run inference with MCTS, simply use the following script. You can serve model using vLLM and update the config file in `Qwen32B_prompt_CFG` and then run the following script.
```python ```python
from coati.reasoner.guided_search.mcts import MCTS from coati.reasoner.guided_search.mcts import MCTS
from coati.reasoner.guided_search.prompt_store.qwen import Qwen32B_prompt_CFG from coati.reasoner.guided_search.prompt_store.qwen import Qwen32B_prompt_CFG