Merge pull request #6482 from ProGamerGov/patch-6

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

View File

@ -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,