[BUG]: No module named 'chatgpt.nn'

[shendlcode]

🐛 Describe the bug

Hi,
I can not run demo of ColossalAI.
The following is the output.

/ColossalAI/dev_test/benchmark_gpt_dummy.py", line 1, in
from chatgpt.nn import GPTActor, GPTCritic, RewardModel
ModuleNotFoundError: No module named 'chatgpt.nn'

Environment

No response

[JThh]

Please follow what's instructed in this file (most importantly, pip install . in the applications/ChatGPT folder.)

[chaos2024]

got the same error, and i already pip install . under the chatgpt folder. I can import chatgpt from python environment, the error comes out with the torchrun command

[A-biao96]

chatgpt.nn is in this path "ColossalAI/applications/ChatGPT/chatgpt/nn", put trian_XX.py with the folder chatgpt, in the same path.

[A-biao96]

得到同样的错误，我已经 pip install .在 chatgpt 文件夹下。我可以从python环境中导入chatgpt，torchrun命令出现错误

You did not successfully install pytorch，please test it with the following code：
import torch
print(torch.cuda.is_available())

[chaos2024]

得到同样的错误，我已经 pip install .在 chatgpt 文件夹下。我可以从python环境中导入chatgpt，torchrun命令出现错误

You did not successfully install pytorch，please test it with the following code： import torch print(torch.cuda.is_available())

thanks

moving file to chatgpt/nn didn't work, the error became no module named transformers while it's been installed.

and torch.cuda.is_available()) is True

[wqw547243068]

Warning:

• chatgpt.nn has been modified as chatgpt.models
• The overall arch diagram is ...

If this error (No module named 'chatgpt.nn') still pops up, remember to git pull in order to update the code

[binmakeswell]

Warning:

• chatgpt.nn has been modified as chatgpt.models
• The overall arch diagram is ...

If this error (No module named 'chatgpt.nn') still pops up, remember to git pull in order to update the code

Thanks for such a clear summary!

[binmakeswell]

We have updated a lot.
Please check the latest instructions
https://github.com/hpcaitech/ColossalAI/tree/main/applications/Chat#install
This issue was closed due to inactivity. Thanks.