Update davit.py

timm/models/davit.py

@@ -560,14 +560,14 @@ class DaViT(nn.Module):
     
     
     def _cfg(url='', **kwargs): # not sure how this should be set up
-    return {
-        'url': url,
-        'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': (7, 7),
-        'crop_pct': 0.875, 'interpolation': 'bilinear',
-        'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
-        'first_conv': 'patch_embeds.0.proj', 'classifier': 'head.fc',
-        **kwargs
-    }
+        return {
+            'url': url,
+            'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': (7, 7),
+            'crop_pct': 0.875, 'interpolation': 'bilinear',
+            'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
+            'first_conv': 'patch_embeds.0.proj', 'classifier': 'head.fc',
+            **kwargs
+        }