Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
sungho.park committed Mar 16, 2023
1 parent 4cc15d2 commit 0479987
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,5 +8,28 @@
- You can use alpaca_data.hf which is converted for using Huggingface Datasets
- Split train and validation for clm training

## Dependency
```sh
pip install -r requirements.txt
```

## Train
```sh
# Use PEFT, LORA
accelerate launch --config_file peft_config.yaml finetune.py

# Use Huggingface Arguments for controll all situations during training.
# You can use train.sh.
# Stil updating...
python train.py \
--model_name_or_path decapoda-research/llama-7b-hf \
--dataset_name alpaca_data.hf \
--is_dataset_from_disk True \
--per_device_train_batch_size 8 \
--per_device_eval_batch_size 8 \
--do_train \
--do_eval \
--output_dir test-clm
```
The codes are still updated, so maybe there're can be some unexpected error.
I used base code from https://github.com/tloen/alpaca-lora. Thanks a lot!!

0 comments on commit 0479987

Please sign in to comment.