From 8792e906abb095ea8df617c5b1bedc80c3483433 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 07:16:33 -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 5fcc6052..1bc8c7b0 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -549,7 +549,7 @@ def _create_davit(variant, pretrained=False, **kwargs):
     model_cls = DaViT
     features_only = False
     kwargs_filter = None
-    if model_kwargs.pop('features_only', False):
+    if kwargs.pop('features_only', False):
         model_cls = DaViTFeatures
         kwargs_filter = ('num_classes', 'global_pool')
         features_only = True