From 37191d8337e023cefddc36dc70e4f797510a5810 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 07:11:03 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index b11342b8..3f0d7107 100644 --- 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