Multi-GPU parallel encoding support for training videos. (#6)

* Multi-GPU parallel encoding support for training videos.

* revert

* make style

* update

---------

Co-authored-by: Aryan <aryan@huggingface.co>
This commit is contained in:
Yuxuan.Zhang
2024-10-09 06:49:59 +08:00
committed by GitHub
parent 3a519f5443
commit cc1d2e759f
5 changed files with 236 additions and 65 deletions
+1 -1
View File
@@ -402,4 +402,4 @@ With `train_batch_size = 4`:
- [x] Make 5B lora finetuning work in under 24GB
> [!IMPORTANT]
> Since our goal is to make the scripts as memory-friendly as possible we don't guarantee multi-GPU training.
> Since our goal is to make the scripts as memory-friendly as possible we don't guarantee multi-GPU training.