Closed CHN-ChenYi closed 2 years ago
Hi @CHN-ChenYi , we have updated the api for the zero init context. Can you install colossalai
from source and try it again? The installation instruction is as follows:
git clone https://github.com/hpcaitech/ColossalAI.git
cd ColossalAI
# install dependency
pip install -r requirements/requirements.txt
# install colossalai
pip install .
Hi @CHN-ChenYi , we have updated the api for the zero init context. Can you install
colossalai
from source and try it again?
Thanks! The problem is solved after I changed to the latest version and rebuilt colossalai.
🐛 Describe the bug
Command:
Result:
Environment
colossalai
nvcc:
Python
PyTorch