Add fallback for Protocol import
This commit is contained in:
parent
151233399c
commit
984b86dd0a
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue