Update docstrings
Browse files- sample_finetune.py +1 -1
sample_finetune.py
CHANGED
|
@@ -22,7 +22,7 @@ Please follow these steps to run the script:
|
|
| 22 |
1. Install dependencies:
|
| 23 |
conda install -c conda-forge accelerate
|
| 24 |
pip3 install -i https://pypi.org/simple/ bitsandbytes
|
| 25 |
-
pip3 install peft
|
| 26 |
pip3 install deepspeed
|
| 27 |
2. Setup accelerate and deepspeed config based on the machine used:
|
| 28 |
accelerate config
|
|
|
|
| 22 |
1. Install dependencies:
|
| 23 |
conda install -c conda-forge accelerate
|
| 24 |
pip3 install -i https://pypi.org/simple/ bitsandbytes
|
| 25 |
+
pip3 install peft transformers trl datasets
|
| 26 |
pip3 install deepspeed
|
| 27 |
2. Setup accelerate and deepspeed config based on the machine used:
|
| 28 |
accelerate config
|