Unverified Commit 89c36630 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #6482 from ProGamerGov/patch-6

Add fallback for Protocol import
parents 49c4509c 984b86dd
Loading
Loading
Loading
Loading
+7 −1
Original line number Diff line number Diff line
@@ -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,