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

6
timm/models/vision_transformer.py
View File

@ -27,7 +27,11 @@ import logging
import math import math
from collections import OrderedDict from collections import OrderedDict
from functools import partial 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
import torch.nn as nn import torch.nn as nn