Skip to content

Commit 8d646f2

Browse files
williambermansayakpaulpcuenca
authored
dreambooth docs torch.compile note (#3471)
* dreambooth docs torch.compile note * Update examples/dreambooth/README.md Co-authored-by: Sayak Paul <[email protected]> * Update examples/dreambooth/README.md Co-authored-by: Pedro Cuenca <[email protected]> --------- Co-authored-by: Sayak Paul <[email protected]> Co-authored-by: Pedro Cuenca <[email protected]>
1 parent 8917769 commit 8d646f2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/dreambooth/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,8 @@ from accelerate.utils import write_basic_config
4343
write_basic_config()
4444
```
4545

46+
When running `accelerate config`, if we specify torch compile mode to True there can be dramatic speedups.
47+
4648
### Dog toy example
4749

4850
Now let's get our dataset. For this example we will use some dog images: https://huggingface.co/datasets/diffusers/dog-example.

0 commit comments

Comments
 (0)