[Fix] fix dreambooth (open-mmlab#1826)

zhuang2002 · Apr 28, 2023 · 9018722 · 9018722
1 parent f252d41
commit 9018722
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/configs/dreambooth/README.md b/configs/dreambooth/README.md
@@ -37,10 +37,11 @@ The file structure will be like this:
 ```text
 data
 └── dreambooth
-    ├── alvan-nee-Id1DBHv4fbg-unsplash.jpeg
-    ├── alvan-nee-bQaAJCbNq3g-unsplash.jpeg
-    ├── alvan-nee-brFsZ7qszSY-unsplash.jpeg
-    └── alvan-nee-eoqnr8ikwFE-unsplash.jpeg
+    └──imgs
+       ├── alvan-nee-Id1DBHv4fbg-unsplash.jpeg
+       ├── alvan-nee-bQaAJCbNq3g-unsplash.jpeg
+       ├── alvan-nee-brFsZ7qszSY-unsplash.jpeg
+       └── alvan-nee-eoqnr8ikwFE-unsplash.jpeg
 ```
 
 2. Start training with the following command:

diff --git a/mmagic/models/utils/model_utils.py b/mmagic/models/utils/model_utils.py
@@ -284,7 +284,7 @@ def set_xformers(module: nn.Module, prefix: str = '') -> nn.Module:
         nn.Module: The module with xformers' efficient Attention.
     """
 
-    if not xformers_is_enable:
+    if not xformers_is_enable():
         print_log('Do not support Xformers. Please install Xformers first. '
                   'The program will run without Xformers.')
         return