Fix errant import

pull/1467/head
Ross Wightman 2 years ago
parent 9709dbaaa9
commit 9d65557be3

@ -13,6 +13,7 @@ except ImportError:
from torch.hub import _get_torch_home as get_dir
from timm import __version__
try:
from huggingface_hub import HfApi, HfFolder, Repository, hf_hub_download, hf_hub_url
hf_hub_download = partial(hf_hub_download, library_name="timm", library_version=__version__)

@ -25,7 +25,6 @@ from functools import partial
from collections import OrderedDict
from typing import Optional
import huggingface_hub.file_download
import torch
import torch.nn as nn
import torch.nn.functional as F

Loading…
Cancel
Save