README.md 439 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
## Overview
This example shows how to use ColossalAI to run huggingface GPT training with Gemini and ZeRO DDP.

## GPT
We use the huggingface transformers GPT2 model. The input data is randonly generated.

## Our Modifications
We adapt the OPT training code to ColossalAI by leveraging Gemini and ZeRO DDP.

## Quick Start
You can launch training by using the following bash script

```bash
pip install -r requirements.txt
bash run.sh
```