open-sora

Runtime error

ver217 commited on Jun 21

Commit

a3f163e

•

1 Parent(s): a0b12a8

Update configs/opensora-v1-2/inference/sample.py

Files changed (1) hide show

configs/opensora-v1-2/inference/sample.py CHANGED Viewed

@@ -17,7 +17,7 @@ model = dict(
     type="STDiT3-XL/2",
     from_pretrained="hpcai-tech/OpenSora-STDiT-v3",
     qk_norm=True,
-    enable_flash_attn=True,
     enable_layernorm_kernel=False,
 )
 vae = dict(

     type="STDiT3-XL/2",
     from_pretrained="hpcai-tech/OpenSora-STDiT-v3",
     qk_norm=True,
+    enable_flash_attn=False,
     enable_layernorm_kernel=False,
 )
 vae = dict(