binmakeswell / ColossalChat

ColossalChat is the project to implement LLM with RLHF, powered by the Colossal-AI project.
https://github.com/hpcaitech/ColossalAI/tree/main/applications/Chat
Apache License 2.0
61 stars 17 forks source link

ModuleNotFoundError: No module named 'loralib' #6

Closed zhoujinzhoujin closed 2 months ago

zhoujinzhoujin commented 2 months ago

I already ran pip install . and tried loralib 0.1.0, 0.1.1 and 0.1.2, but still see ModuleNotFoundError:

(coati) (base) jinzhou@jupiter:~/ColossalChat/examples$ conda list loralib
# packages in environment at /home/jinzhou/miniconda3/envs/coati:
#
# Name                    Version                   Build  Channel
loralib                   0.1.1                    pypi_0    pypi
(coati) (base) jinzhou@jupiter:~/ColossalChat/examples$ ./train_sft.sh
master_addr is only used for static rdzv_backend and when rdzv_endpoint is not specified.
WARNING:torch.distributed.run:
*****************************************
Setting OMP_NUM_THREADS environment variable for each process to be 1 in default, to avoid your system being overloaded, please further tune the variable for optimal performance in your application as needed. 
*****************************************
Traceback (most recent call last):
  File "/home/jinzhou/ColossalChat/examples/train_sft.py", line 4, in <module>
    import loralib as lora
ModuleNotFoundError: No module named 'loralib'
Traceback (most recent call last):
  File "/home/jinzhou/ColossalChat/examples/train_sft.py", line 4, in <module>
    import loralib as lora
ModuleNotFoundError: No module named 'loralib'
Traceback (most recent call last):
  File "/home/jinzhou/ColossalChat/examples/train_sft.py", line 4, in <module>
    import loralib as lora
ModuleNotFoundError: No module named 'loralib'
Traceback (most recent call last):
  File "/home/jinzhou/ColossalChat/examples/train_sft.py", line 4, in <module>
    import loralib as lora
ModuleNotFoundError: No module named 'loralib'

How to fix this problem?