diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-09 20:06:50 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-09 20:06:50 +0300 |
| commit | 89c3663080658b84b506c12563a729b6fc65ae10 (patch) | |
| tree | 0b593b2c7907070ba0511806643155441e2838ca /modules/sub_quadratic_attention.py | |
| parent | 49c4509ce2302350210ff650fd26373518c46a79 (diff) | |
| parent | 984b86dd0abf0da7f6b116864c791a2bfe8859ef (diff) | |
Merge pull request #6482 from ProGamerGov/patch-6
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, |