refactor
- unify all pipelines into one - unify transformer model into one - separate VAE - add single file model loading
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user