add sageattention

Signed-off-by: Vladimir Mandic <mandic00@live.com>
2026-09-19 17:24:32 +02:00 · 2024-10-12 15:43:34 -04:00
parent 73bd0816d1
commit 0c54c235cb
4 changed files with 17 additions and 10 deletions
@@ -1,9 +1,8 @@
 import os
-from functools import wraps
+from functools import wraps, cache
 import torch
 import diffusers #0.29.1 # pylint: disable=import-error
 from diffusers.models.attention_processor import Attention
-from functools import cache

 # pylint: disable=protected-access, missing-function-docstring, line-too-long