diff --git a/backend/attention.py b/backend/attention.py index f6cc1ec1..60afe25f 100644 --- a/backend/attention.py +++ b/backend/attention.py @@ -1,8 +1,6 @@ import torch from backend.args import args -from einops import rearrange, repeat -from typing import Optional if args.xformers: import xformers