diff options
| author | ProGamerGov <ProGamerGov@users.noreply.github.com> | 2023-01-07 20:08:21 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-07 20:08:21 +0000 |
| commit | 984b86dd0abf0da7f6b116864c791a2bfe8859ef (patch) | |
| tree | 4db1ed2c30c98c572aa2a5308e329d474d6f6557 /modules/sub_quadratic_attention.py | |
| parent | 151233399c4b79934bdbb7c12a97eeb6499572fb (diff) | |
| download | stable-diffusion-webui-gfx803-984b86dd0abf0da7f6b116864c791a2bfe8859ef.tar.gz stable-diffusion-webui-gfx803-984b86dd0abf0da7f6b116864c791a2bfe8859ef.tar.bz2 stable-diffusion-webui-gfx803-984b86dd0abf0da7f6b116864c791a2bfe8859ef.zip | |
Add fallback for Protocol import
Diffstat (limited to 'modules/sub_quadratic_attention.py')
| -rw-r--r-- | modules/sub_quadratic_attention.py | 8 |
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, |