Update davit.py

3 years ago · 37191d8337
parent 61a792801c
commit 37191d8337
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -14,7 +14,7 @@ DaViT model defs and weights adapted from https://github.com/dingmyu/davit, orig

 import itertools

-from typing import Any, Dict, Iterable, Iterator, Mapping, Optional, overload, Tuple, TypeVar, Union
+from typing import Any, Dict, Iterable, Iterator, Mapping, Optional, overload, Tuple, TypeVar, Union, List