search

huggingface / alignment-handbook

Robust recipes to align language models with human and AI preferences
https://huggingface.co/HuggingFaceH4
Apache License 2.0
4.45k stars 385 forks source link

Running on single RTX4090 #23

Closed jasonwtli closed 9 months ago

jasonwtli commented 9 months ago

Hi,

Followed the install. Tried to run in shell # Step 1 - SFT. for single gpu as per instructions. i.e --num_processes=1

Kicked up the below error. Is alignment module something we need to seperately pip install?

''' from alignment import ( ModuleNotFoundError: No module named 'alignment'

'''

Thanks in advance! Jason.

edbeeching commented 9 months ago

Yes, please follow the installation instructions here. I will close the issue but let us know if you have further difficulties.