aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-01-09 17:06:50 +0000
committerGitHub <noreply@github.com>2023-01-09 17:06:50 +0000
commit89c3663080658b84b506c12563a729b6fc65ae10 (patch)
tree0b593b2c7907070ba0511806643155441e2838ca
parent49c4509ce2302350210ff650fd26373518c46a79 (diff)
parent984b86dd0abf0da7f6b116864c791a2bfe8859ef (diff)
downloadstable-diffusion-webui-gfx803-89c3663080658b84b506c12563a729b6fc65ae10.tar.gz
stable-diffusion-webui-gfx803-89c3663080658b84b506c12563a729b6fc65ae10.tar.bz2
stable-diffusion-webui-gfx803-89c3663080658b84b506c12563a729b6fc65ae10.zip
Merge pull request #6482 from ProGamerGov/patch-6
Add fallback for Protocol import
-rw-r--r--modules/sub_quadratic_attention.py8
1 files changed, 7 insertions, 1 deletions
diff --git a/modules/sub_quadratic_attention.py b/modules/sub_quadratic_attention.py
index fea7aaac..93381bae 100644
--- a/modules/sub_quadratic_attention.py
+++ b/modules/sub_quadratic_attention.py
@@ -15,7 +15,13 @@ import torch
from torch import Tensor
from torch.utils.checkpoint import checkpoint
import math
-from typing import Optional, NamedTuple, Protocol, List
+
+try:
+ from typing import Protocol
+except:
+ from typing_extensions import Protocol
+
+from typing import Optional, NamedTuple, List
def narrow_trunc(
input: Tensor,