Skip to content

Commit b07c6a4

Browse files
authored
Merge branch 'main' into xpu-enabling2
2 parents 6238972 + 464374f commit b07c6a4

File tree

7 files changed

+535
-145
lines changed

7 files changed

+535
-145
lines changed

examples/dreambooth/train_dreambooth_lora_sana.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -995,7 +995,8 @@ def main(args):
995995
if args.enable_npu_flash_attention:
996996
if is_torch_npu_available():
997997
logger.info("npu flash attention enabled.")
998-
transformer.enable_npu_flash_attention()
998+
for block in transformer.transformer_blocks:
999+
block.attn2.set_use_npu_flash_attention(True)
9991000
else:
10001001
raise ValueError("npu flash attention requires torch_npu extensions and is supported only on npu device ")
10011002

examples/model_search/README.md

Lines changed: 2 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -82,31 +82,11 @@ pipeline = EasyPipelineForInpainting.from_huggingface(
8282
## Search Civitai and Huggingface
8383

8484
```python
85-
from pipeline_easy import (
86-
search_huggingface,
87-
search_civitai,
88-
)
89-
90-
# Search Lora
91-
Lora = search_civitai(
92-
"Keyword_to_search_Lora",
93-
model_type="LORA",
94-
base_model = "SD 1.5",
95-
download=True,
96-
)
9785
# Load Lora into the pipeline.
98-
pipeline.load_lora_weights(Lora)
99-
86+
pipeline.auto_load_lora_weights("Detail Tweaker")
10087

101-
# Search TextualInversion
102-
TextualInversion = search_civitai(
103-
"EasyNegative",
104-
model_type="TextualInversion",
105-
base_model = "SD 1.5",
106-
download=True
107-
)
10888
# Load TextualInversion into the pipeline.
109-
pipeline.load_textual_inversion(TextualInversion, token="EasyNegative")
89+
pipeline.auto_load_textual_inversion("EasyNegative", token="EasyNegative")
11090
```
11191

11292
### Search Civitai

0 commit comments

Comments
 (0)