From 956d7e66d791e9c35da3f5cf42b79cd59d5d4c4f Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Sat, 8 Jun 2024 00:00:42 -0700
Subject: [PATCH] Update cvt.py

---
 timm/models/cvt.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/timm/models/cvt.py b/timm/models/cvt.py
index 95f2f9d0d5..b47e15b1d9 100644
--- a/timm/models/cvt.py
+++ b/timm/models/cvt.py
@@ -389,6 +389,7 @@ def __init__(
         mlp_ratio: float = 4.,
         mlp_act_layer: nn.Module = QuickGELU,
         use_cls_token: Tuple[bool, ...] = (False, False, True),
+        drop_rate: float = 0.,
     ) -> None:
         super().__init__()
         num_stages = len(dims)