ACE-Plus

Running on Zero

chaojiemao commited on Dec 8, 2024

Commit

c1ad6f3

verified ·

1 Parent(s): 08e3efa

Update config/models/ace_flux_dev.yaml

Files changed (1) hide show

config/models/ace_flux_dev.yaml CHANGED Viewed

@@ -113,7 +113,7 @@ MODEL:
     DEPTH: 19
     # DEPTH_SINGLE_BLOCKS DESCRIPTION: number of transformer blocks in the single stream block. TYPE: int default: 38
     DEPTH_SINGLE_BLOCKS: 38
-    #ATTN_BACKEND: flash_attn
   #
   FIRST_STAGE_MODEL:
     NAME: AutoencoderKLFlux

     DEPTH: 19
     # DEPTH_SINGLE_BLOCKS DESCRIPTION: number of transformer blocks in the single stream block. TYPE: int default: 38
     DEPTH_SINGLE_BLOCKS: 38
+    ATTN_BACKEND: flash_attn
   #
   FIRST_STAGE_MODEL:
     NAME: AutoencoderKLFlux