michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request #6482 from ProGamerGov/patch-6

Browse source 
Add fallback for Protocol import
This commit is contained in:
AUTOMATIC1111 2023-01-09 20:06:50 +03:00 committed by GitHub
commit 89c3663080
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
modules/sub_quadratic_attention.py
View file

@ -15,7 +15,13 @@ import torch
from torch import Tensor from torch import Tensor
from torch.utils.checkpoint import checkpoint from torch.utils.checkpoint import checkpoint
import math 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( def narrow_trunc(
input: Tensor, input: Tensor,