Unverified Commit 8d646f22 authored by Will Berman's avatar Will Berman Committed by GitHub
Browse files

dreambooth docs torch.compile note (#3471)



* dreambooth docs torch.compile note

* Update examples/dreambooth/README.md
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>

* Update examples/dreambooth/README.md
Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>

---------
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>
parent 89177694
...@@ -43,6 +43,8 @@ from accelerate.utils import write_basic_config ...@@ -43,6 +43,8 @@ from accelerate.utils import write_basic_config
write_basic_config() write_basic_config()
``` ```
When running `accelerate config`, if we specify torch compile mode to True there can be dramatic speedups.
### Dog toy example ### Dog toy example
Now let's get our dataset. For this example we will use some dog images: https://huggingface.co/datasets/diffusers/dog-example. Now let's get our dataset. For this example we will use some dog images: https://huggingface.co/datasets/diffusers/dog-example.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment