mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Literal use w/ python < 3.8 requires typing_extension, cach instead of check sys ver

Browse Source
This commit is contained in:
Ross Wightman 2023-11-21 08:22:33 -08:00 committed by Ross Wightman
parent dfaab97d20
commit ada145b016
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
timm/models/_hub.py
View File

@ -2,7 +2,6 @@ import hashlib
import json
import logging
import os
import sys
from functools import partial
from pathlib import Path
from tempfile import TemporaryDirectory
@ -22,9 +21,9 @@ try:
except ImportError:
_has_safetensors = False
if sys.version_info >= (3, 8):
try:
from typing import Literal
else:
except ImportError:
from typing_extensions import Literal
from timm import __version__

6
timm/models/vision_transformer.py
View File

@ -27,7 +27,11 @@ import logging
import math
from collections import OrderedDict
from functools import partial
from typing import Any, Callable, Dict, Optional, Sequence, Set, Tuple, Type, Union, Literal, List
from typing import Any, Callable, Dict, Optional, Sequence, Set, Tuple, Type, Union, List
try:
from typing import Literal
except ImportError:
from typing_extensions import Literal
import torch
import torch.nn as nn