Update README.md

This commit is contained in:
Yu Meng
2024-05-24 10:28:52 -04:00
committed by GitHub
parent a53eb35980
commit e80a0fc4f4
+1 -1
View File
@@ -77,7 +77,7 @@ python -m pip install flash-attn --no-build-isolation
## Training Scripts
We provide four training config files for the four training setups reported in our paper:
We provide four training config files for the four training setups reported in our paper. The training config is set for 8xH100 GPUs. You may need to adjust `num_processes` and `per_device_train_batch_size` based on your computation environment.
* Mistral-Base:
```shell