- unify all pipelines into one
- unify transformer model into one
- separate VAE
- add single file model loading
This commit is contained in:
kijai
2024-11-19 03:04:22 +02:00
parent 6f9e4ff647
commit a7646c0d6f
13 changed files with 594 additions and 5345 deletions
-1
View File
@@ -76,7 +76,6 @@ class CogVideoXAttnProcessor2_0:
if not hasattr(F, "scaled_dot_product_attention"):
raise ImportError("CogVideoXAttnProcessor requires PyTorch 2.0, to use it, please upgrade PyTorch to 2.0.")
#@torch.compiler.disable()
def __call__(
self,
attn: Attention,