open-sora

Runtime error

ver217 commited on Jun 21

Commit

a0b12a8

•

1 Parent(s): e8b9ffe

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

Files changed (1) hide show

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

@@ -18,7 +18,7 @@ model = dict(
     from_pretrained="hpcai-tech/OpenSora-STDiT-v3",
     qk_norm=True,
     enable_flash_attn=True,
-    enable_layernorm_kernel=True,
 )
 vae = dict(
     type="OpenSoraVAE_V1_2",

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