Fix errant import

huggingface · Sep 16, 2022 · 9d65557 · 9d65557
1 parent 9709dba
commit 9d65557
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/timm/models/hub.py b/timm/models/hub.py
@@ -13,6 +13,7 @@
     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__)

diff --git a/timm/models/vision_transformer.py b/timm/models/vision_transformer.py
@@ -25,7 +25,6 @@
 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