From 8d646f229440999f8c20bf8cbaf016dc4b35441d Mon Sep 17 00:00:00 2001 From: Will Berman Date: Thu, 18 May 2023 19:10:14 -0700 Subject: [PATCH] dreambooth docs torch.compile note (#3471) * dreambooth docs torch.compile note * Update examples/dreambooth/README.md Co-authored-by: Sayak Paul * Update examples/dreambooth/README.md Co-authored-by: Pedro Cuenca --------- Co-authored-by: Sayak Paul Co-authored-by: Pedro Cuenca --- examples/dreambooth/README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/dreambooth/README.md b/examples/dreambooth/README.md index 086100bd4a..83073210ac 100644 --- a/examples/dreambooth/README.md +++ b/examples/dreambooth/README.md @@ -43,6 +43,8 @@ from accelerate.utils import 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 Now let's get our dataset. For this example we will use some dog images: https://huggingface.co/datasets/diffusers/dog-example.