Skip to content

Commit 4e3cc2c

Browse files
author
蒋硕
committed
Improve the performance and suitable for NPU
1 parent 8694715 commit 4e3cc2c

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

examples/text_to_image/train_text_to_image_sdxl.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,6 @@
6161
if is_torch_npu_available():
6262
import torch_npu
6363
torch.npu.config.allow_internal_format = False
64-
torch.npu.set_compile_mode(jit_compile=False)
6564

6665
DATASET_NAME_MAPPING = {
6766
"lambdalabs/naruto-blip-captions": ("image", "text"),

0 commit comments

Comments
 (0)