Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to use it. Is there some code examples? #28

Open
Mr-IT007 opened this issue May 16, 2023 · 1 comment
Open

How to use it. Is there some code examples? #28

Mr-IT007 opened this issue May 16, 2023 · 1 comment

Comments

@Mr-IT007
Copy link

How to use it. Is there some code examples?

@GanjinZero
Copy link
Owner

For training:

Training with RRHF
You can train your own model with generated or released datasets using the script train.sh, please note that the training process requires 8*A100 80GB GPUs, bf16 and FSDP. In the future, we will try efficient training methods such as LoRA or Prefix-tuning or Adapter to lower the computational resource requirements.

bash ./train.sh <path_to_alpaca_directory> <save_path_directory> <path_to_data_json>

For using Wombat:
Use recover_wombat_7b.sh and single_sentence_inference.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants